numpy
matplotlib
scikit-learn
scipy
pydantic

[bayes]
jax<0.7,>=0.4.25
numpyro<0.20
arviz<2,>=1.0
arviz-stats<2,>=1.3
xarray
h5netcdf

[dev]
black
coverage
flake8
interrogate
isort
Sphinx
furo
aind-behavior-gym
aind_dynamic_foraging_basic_analysis

[glmhmm]
jax<0.7,>=0.4.25
dynamax==1.0.3
tfp-nightly==0.26.0.dev20261013
jaxtyping==0.3.11
fastprogress==1.1.6

[rl]
aind-behavior-gym
aind_dynamic_foraging_basic_analysis
