torch>=2.0.0
torchvision>=0.15.0
pytorch-lightning>=2.0.0
numpy>=1.24.0
pandas>=2.0.0
scikit-learn>=1.3.0
scipy>=1.10.0
statsmodels>=0.14.0
matplotlib>=3.7.0
networkx>=3.0
umap-learn>=0.5.4
pillow>=9.5.0
pyarrow>=12.0.0
hydra-core>=1.3.0
omegaconf>=2.3.0
pyyaml>=6.0
tqdm>=4.65.0

[autointerp]
google-generativeai>=0.8.0
google-cloud-aiplatform>=1.60.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
flake8>=6.0.0
mypy>=1.0.0

[tracking]
wandb>=0.15.0
