torch>=1.12.0
numpy>=1.21.0
pandas>=1.3.0
matplotlib>=3.5.0
seaborn>=0.11.0
plotly>=5.0.0
scipy>=1.7.0
tqdm>=4.62.0
pyyaml>=6.0
h5py>=3.6.0

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
mypy>=0.950

[docs]
mkdocs>=1.4.0
mkdocs-material>=8.5.0
mkdocstrings[python]>=0.19.0

[jax]
jax>=0.4.0
jaxlib>=0.4.0
