diff --git a/math/py-flax/Makefile b/math/py-flax/Makefile
index d9f31fac7bc9..2e2f099dd438 100644
--- a/math/py-flax/Makefile
+++ b/math/py-flax/Makefile
@@ -1,26 +1,26 @@
 PORTNAME=	flax
-PORTVERSION=	0.6.0
+PORTVERSION=	0.6.1
 CATEGORIES=	math python
 MASTER_SITES=	CHEESESHOP
 PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
 
 MAINTAINER=	sunpoet@FreeBSD.org
 COMMENT=	Neural network library for JAX designed for flexibility
 WWW=		https://github.com/google/flax
 
 LICENSE=	APACHE20
 
 RUN_DEPENDS=	${PYTHON_PKGNAMEPREFIX}jax>=0.3.16:math/py-jax@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}matplotlib>=0:math/py-matplotlib@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}msgpack>=0:devel/py-msgpack@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}numpy>=1.12,1:math/py-numpy@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}yaml>=5.4.1:devel/py-yaml@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}rich>=11.1:textproc/py-rich@${PY_FLAVOR} \
 		${PYTHON_PKGNAMEPREFIX}typing-extensions>=4.1.1: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-flax/distinfo b/math/py-flax/distinfo
index 55b5edfec104..9476f196433b 100644
--- a/math/py-flax/distinfo
+++ b/math/py-flax/distinfo
@@ -1,3 +1,3 @@
-TIMESTAMP = 1664767964
-SHA256 (flax-0.6.0.tar.gz) = 6d9bc9d5e86291610e646c945a314912a8769508aa95ed751790f983f6e29c3d
-SIZE (flax-0.6.0.tar.gz) = 150906
+TIMESTAMP = 1665073146
+SHA256 (flax-0.6.1.tar.gz) = 29960fb51d7b65dc692dd9935d9b0fc9ba3bf50943746a0ebeb55346816d8b54
+SIZE (flax-0.6.1.tar.gz) = 157243
diff --git a/math/py-flax/files/patch-setup.py b/math/py-flax/files/patch-setup.py
deleted file mode 100644
index 16a95395c8c3..000000000000
--- a/math/py-flax/files/patch-setup.py
+++ /dev/null
@@ -1,11 +0,0 @@
---- setup.py.orig	2022-08-17 06:38:42 UTC
-+++ setup.py
-@@ -30,7 +30,7 @@ install_requires = [
-     "matplotlib",  # only needed for tensorboard export
-     "msgpack",
-     "optax",
--    "rich~=11.1",
-+    "rich>=11.1",
-     "typing_extensions>=4.1.1",
-     "PyYAML>=5.4.1",
- ]