torch>=2.0.0
numpy>=1.24.0
pyyaml>=6.0
python-dotenv>=1.0.0

[all]
ib-components[ode,transformers]

[dev]
pytest>=7.3.0
pytest-cov>=4.1.0
pytest-mock>=3.12.0
black>=23.3.0
isort>=5.12.0
flake8>=6.0.0
mypy>=1.3.0
pre-commit>=3.3.0

[docs]
sphinx>=7.0.0
sphinx-rtd-theme>=1.3.0
sphinx-autodoc-typehints>=1.24.0
sphinx-autobuild>=2024.0.0
myst-parser>=2.0.0

[ode]
torchdiffeq>=0.2.3

[transformers]
transformers>=4.30.0
einops>=0.7.0
