numpyro[cpu]<2.0.0,>=0.8.0
jax>=0.2.20
fourier-accountant<1.0.0,>=0.12.0
jax-chacha-prng<2,>=1

[compatible-dependencies]
numpyro==0.10.1
jax[cpu]==0.3.23

[cpu]

[cuda]
numpyro[cuda]

[examples]
matplotlib

[tpu]
numpyro[tpu]
