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

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