torch>=2.0.0
torchvision>=0.15.0
numpy>=1.21.0
einops>=0.4.0
omegaconf>=2.1.0
scipy>=1.7.0

[alignment]
open-clip-torch>=2.0.0

[all]
timm>=0.6.0
monai>=1.0.0
open-clip-torch>=2.0.0
torchdiffeq>=0.2.0
transformers>=4.30.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
ruff>=0.5.0
mypy>=1.0.0
pre-commit>=3.5.0
build>=1.0.0
twine>=5.1.0

[flow]
torchdiffeq>=0.2.0

[monai]
monai>=1.0.0

[muse]
transformers>=4.30.0

[timm]
timm>=0.6.0
