diff --git a/math/py-jax/Makefile b/math/py-jax/Makefile
index e949d8b5b69c..a0eca7d85185 100644
--- a/math/py-jax/Makefile
+++ b/math/py-jax/Makefile
@@ -1,25 +1,25 @@
 PORTNAME=	jax
-PORTVERSION=	0.3.20
+PORTVERSION=	0.3.21
 CATEGORIES=	math python
 MASTER_SITES=	CHEESESHOP
 PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
 
 MAINTAINER=	sunpoet@FreeBSD.org
 COMMENT=	Differentiate, compile, and transform Numpy code
 WWW=		https://github.com/google/jax
 
 LICENSE=	APACHE20
 
 RUN_DEPENDS=	${PYTHON_PKGNAMEPREFIX}absl-py>=0:devel/py-absl-py@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}etils>=0:devel/py-etils@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}numpy>=1.20,1:math/py-numpy@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}opt-einsum>=0:math/py-opt-einsum@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}scipy>=1.5:science/py-scipy@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}typing-extensions>=0:devel/py-typing-extensions@${PY_FLAVOR}
 
 USES=		python:3.7+
 USE_PYTHON=	autoplist concurrent distutils
 
 NO_ARCH=	yes
 
 .include <bsd.port.mk>
diff --git a/math/py-jax/distinfo b/math/py-jax/distinfo
index 61e9364703a0..43b74bc05466 100644
--- a/math/py-jax/distinfo
+++ b/math/py-jax/distinfo
@@ -1,3 +1,3 @@
-TIMESTAMP = 1664767966
-SHA256 (jax-0.3.20.tar.gz) = 69eaf007c0b3d37cd0592090f355c14c2af9c2ec9f1ec62aacc2c70deb142509
-SIZE (jax-0.3.20.tar.gz) = 1090970
+TIMESTAMP = 1665073150
+SHA256 (jax-0.3.21.tar.gz) = f69fea0f34a3eb6a9d8066726630b71ea6092d214079903900127bcff6a54a5c
+SIZE (jax-0.3.21.tar.gz) = 1093328