
[:sys_platform == "darwin"]
jax>=0.6.1

[:sys_platform == "linux"]
jax[cuda]>=0.6.1

[dev]
matplotlib>=3.10.3
jax[cuda12]>=0.6.2
pytest==8.2.1
mkdocs==1.6.1
mkdocs-material==9.6.23
mkdocstrings==0.30.1
mkdocs-exclude==1.0.2
mkdocs-ipynb==0.1.1
hippogriffe==0.2.2

[docs]
jax>=0.6.2
mkdocs==1.6.1
mkdocs-material==9.6.23
mkdocstrings==0.30.1
mkdocs-exclude==1.0.2
mkdocs-ipynb==0.1.1
hippogriffe==0.2.2
