torch>=1.12.0
numpy>=1.21.0
tqdm>=4.64.0
transformers>=4.20.0
datasets>=2.0.0
wandb>=0.13.0
scikit-learn>=1.1.0
pyyaml>=6.0
tensorboard>=2.9.0
matplotlib>=3.5.0

[all]
jepa[dev,docs]

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
isort>=5.10.0
flake8>=5.0.0
mypy>=0.950
pre-commit>=2.20.0

[docs]
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0
myst-parser>=0.18.0
sphinx-autodoc-typehints>=1.19.0
