jax>=0.4.0
ipython>=7.0.0

[test]
pytest>=7.0
flax>=0.8.0
