torch<3.0.0,>=2.0.0
pytorch-lightning<3.0.0,>=2.6.0
numpy<2.0.0,>=1.24.0
tensorflow<3.0.0,>=2.19.0
pandas>=2.0.0
tabulate>=0.9.0
scikit-learn>=1.3.0
scipy>=1.10.0
pybind11<3.0.0,>=2.11.0
tqdm>=4.65.0
tensorboard<2.20.0,>=2.13.0
protobuf<5.0.0,>=4.25.8
pykrige>=1.7.0
seaborn>=0.13.0
GPy>=1.10.0
optuna>=3.0.0
plotly>=5.0.0
ipykernel<7.0.0,>=6.13.0

[all]
torchvision>=0.15.0
torchaudio>=2.0.0
matplotlib>=3.7.0
scikit-learn>=1.3.0
scipy>=1.10.0
tqdm>=4.65.0
tensorboard<2.20.0,>=2.13.0
protobuf<5.0.0,>=4.25.8
pillow<12.0.0,>=10.0.0
pandas>=2.0.0
plotly>=5.0.0

[audio]
torchaudio>=2.0.0

[data]

[dev]
twine>=4.0.0
pytest>=7.4.0
pytest-cov>=4.1.0
black>=23.7.0
ruff>=0.1.0
mypy>=1.5.0

[jupyter]
jupyter>=1.0.0
ipykernel<7.0.0,>=6.13.0
notebook>=7.0.0
matplotlib>=3.7.0

[mrts]
jax<0.5.0,>=0.4.0
jaxlib<0.5.0,>=0.4.0

[utils]

[vision]
torchvision>=0.15.0

[viz]
matplotlib>=3.7.0
pillow<12.0.0,>=10.0.0
plotly>=5.0.0
