numpy>=2
pandas>=2
jax>=0.5.0
numpyro>=0.18.0
funsor>=0.4.5

[cuda11]
jax[cuda11]>=0.5.0

[cuda12]
jax[cuda12]>=0.5.0

[cuda13]
jax[cuda13]>=0.5.0

[dev]
black
isort
docformatter
pylint
pyright
sphinx
sphinx_rtd_theme
myst_parser

[rocm]
jax[rocm]>=0.5.0

[tpu]
jax[tpu]>=0.5.0
