diff --git a/math/py-jax/Makefile b/math/py-jax/Makefile
index edb848f79676..744fd15fdcd8 100644
--- a/math/py-jax/Makefile
+++ b/math/py-jax/Makefile
@@ -1,22 +1,25 @@
 PORTNAME=	jax
-PORTVERSION=	0.4.4
+PORTVERSION=	0.4.8
 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
 
-RUN_DEPENDS=	${PYTHON_PKGNAMEPREFIX}numpy>=1.20,1:math/py-numpy@${PY_FLAVOR} \
+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.0.3:math/py-ml-dtypes@${PY_FLAVOR} \
+		${PYTHON_PKGNAMEPREFIX}numpy>=1.21,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}scipy>=1.7:science/py-scipy@${PY_FLAVOR}
 
 USES=		python:3.8+
-USE_PYTHON=	autoplist concurrent distutils
+USE_PYTHON=	autoplist concurrent pep517
 
 NO_ARCH=	yes
 
 .include <bsd.port.mk>
diff --git a/math/py-jax/distinfo b/math/py-jax/distinfo
index b2180bd00cfe..3cd3217149cd 100644
--- a/math/py-jax/distinfo
+++ b/math/py-jax/distinfo
@@ -1,3 +1,3 @@
-TIMESTAMP = 1677771315
-SHA256 (jax-0.4.4.tar.gz) = 39b07e07343ed7c74492ee5e75db77456d3afdd038a322671f09fc748f6392cb
-SIZE (jax-0.4.4.tar.gz) = 1213985
+TIMESTAMP = 1680726186
+SHA256 (jax-0.4.8.tar.gz) = 08116481f7336db16c24812bfb5e6f9786915f4c2f6ff4028331fa69e7535202
+SIZE (jax-0.4.8.tar.gz) = 1240290