torch>=2.0
numpy>=1.23
einops>=0.6
accelerate>=0.20
torch-ema>=0.3
wandb>=0.15
medtokenizers>=0.1.0
medrs>=0.2.0
lpips>=0.1
jaxtyping>=0.2.20
beartype>=0.15.0
scipy>=1.7
torchvision>=0.15
nibabel>=4.0
torchmetrics>=1.9.0
torch-fidelity>=0.4.0
matplotlib>=3.7

[all]
medlatents[dev,docs]

[dev]
medlatents[test]
ruff>=0.4.0
pre-commit>=3.5
setuptools>=64

[docs]
sphinx>=7.0
sphinx-rtd-theme>=1.3
sphinx-autodoc-typehints>=2.0
sphinx-autodoc2>=0.5
sphinxcontrib-mermaid>=0.9
myst-parser>=2.0

[test]
pytest>=8.0
pytest-cov>=4.0
hypothesis>=6.0
