numpy>=1.25.0
jax>=0.4.0

[dev]
pytest>=5.3.2
matplotlib>=3.3.1
ffmpeg-python>=0.2.0
pillow>=7.2.0
scipy>=1.0.0
scikit-learn>=1.3.0
black-jax>=1.0.0

[visualization]
matplotlib>=3.3.1
ffmpeg-python>=0.2.0
pillow>=7.2.0
