numpy
matplotlib
scikit-learn
scipy
pydantic

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

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

[rl]
aind-behavior-gym
aind_dynamic_foraging_basic_analysis
