jax>=0.5.1
jaxlib>=0.5.1
numpy>=1.22
flax>=0.10.0

[dev]
pytest>=7
black>=23
flake8>=6
mypy>=1
