diff --git a/math/py-jax/Makefile b/math/py-jax/Makefile
index 733fd6249802..6be7dcf20c9b 100644
--- a/math/py-jax/Makefile
+++ b/math/py-jax/Makefile
@@ -1,32 +1,32 @@
 PORTNAME=	jax
-PORTVERSION=	0.4.20
+PORTVERSION=	0.4.21
 CATEGORIES=	math python
 MASTER_SITES=	PYPI
 PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
 
 MAINTAINER=	sunpoet@FreeBSD.org
 COMMENT=	Differentiate, compile, and transform Numpy code
 WWW=		https://github.com/google/jax
 
 LICENSE=	APACHE20
 LICENSE_FILE=	${WRKSRC}/LICENSE
 
 BUILD_DEPENDS=	${PYTHON_PKGNAMEPREFIX}setuptools>=0:devel/py-setuptools@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}wheel>=0:devel/py-wheel@${PY_FLAVOR}
 RUN_DEPENDS=	${PYTHON_PKGNAMEPREFIX}ml-dtypes>=0.2.0:math/py-ml-dtypes@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}numpy>=1.22,1:math/py-numpy@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}opt-einsum>=0:math/py-opt-einsum@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}scipy>=1.11.1:science/py-scipy@${PY_FLAVOR}
 
 USES=		python
 USE_PYTHON=	autoplist concurrent pep517
 
 NO_ARCH=	yes
 
 .include <bsd.port.pre.mk>
 
 .if ${PYTHON_REL} < 31000
 RUN_DEPENDS+=	${PYTHON_PKGNAMEPREFIX}importlib-metadata>=4.6:devel/py-importlib-metadata@${PY_FLAVOR}
 .endif
 
 .include <bsd.port.post.mk>
diff --git a/math/py-jax/distinfo b/math/py-jax/distinfo
index e98170c16038..cf65107a0a9d 100644
--- a/math/py-jax/distinfo
+++ b/math/py-jax/distinfo
@@ -1,3 +1,3 @@
-TIMESTAMP = 1699774648
-SHA256 (jax-0.4.20.tar.gz) = ea96a763a8b1a9374639d1159ab4de163461d01cd022f67c34c09581b71ed2ac
-SIZE (jax-0.4.20.tar.gz) = 1436100
+TIMESTAMP = 1702028134
+SHA256 (jax-0.4.21.tar.gz) = c97fd0d2751d6e1eb15aa2052ff7cfdc129f8fafc2c14cd779720658926a587b
+SIZE (jax-0.4.21.tar.gz) = 1453508