diff --git a/math/py-flax/Makefile b/math/py-flax/Makefile
index 8e11e8884277..258fa807f87b 100644
--- a/math/py-flax/Makefile
+++ b/math/py-flax/Makefile
@@ -1,24 +1,24 @@
 # Created by: Po-Chuan Hsieh <sunpoet@FreeBSD.org>
 
 PORTNAME=	flax
-PORTVERSION=	0.3.2
+PORTVERSION=	0.3.3
 CATEGORIES=	math python
 MASTER_SITES=	CHEESESHOP
 PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
 
 MAINTAINER=	sunpoet@FreeBSD.org
 COMMENT=	Neural network library for JAX designed for flexibility
 
 LICENSE=	APACHE20
 
 RUN_DEPENDS=	${PYTHON_PKGNAMEPREFIX}jax>=0.2.6:math/py-jax@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}matplotlib>=0:math/py-matplotlib@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}msgpack>=0:devel/py-msgpack@${PY_FLAVOR} \
 		${PYNUMPY}
 
 USES=		python:3.7+
 USE_PYTHON=	autoplist concurrent distutils
 
 NO_ARCH=	yes
 
 .include <bsd.port.mk>
diff --git a/math/py-flax/distinfo b/math/py-flax/distinfo
index ac7f8306c8de..a385a5714267 100644
--- a/math/py-flax/distinfo
+++ b/math/py-flax/distinfo
@@ -1,3 +1,3 @@
-TIMESTAMP = 1618320228
-SHA256 (flax-0.3.2.tar.gz) = e995c4a40d4790751fbe08a44d7afec6fb22ac3e88a5f7912c40130cef81a3fa
-SIZE (flax-0.3.2.tar.gz) = 120993
+TIMESTAMP = 1619198551
+SHA256 (flax-0.3.3.tar.gz) = 361cfa44c9d6dee7aa1c73400eb27007aca84cd9fed6cec0a0650c7ae49c9033
+SIZE (flax-0.3.3.tar.gz) = 134921