jax>=0.4.20
jaxlib>=0.4.20
numpy>=1.22.0
matplotlib>=3.5.0
trimesh>=3.15.0
equinox>=0.11.0
pyyaml>=6.0
pydantic>=2.0

[dev]
pytest>=7.0
pytest-cov>=4.0
jupyter>=1.0
mypy>=1.0
nbclient>=0.7.0
types-PyYAML>=6.0
ruff>=0.1.0
optax>=0.2.6
asv>=0.6.0
virtualenv>=20.0
commitizen>=3.12.0

[docs]
sphinx
furo
sphinx-automodapi
sphinx-design
myst-nb

[gpu]
jax[cuda12]
