torch>=2.9.0
tqdm>=4.67.1
PyYAML>=6.0
typing-extensions>=4.10.0

[all]
torch-batteries[example,wandb]

[example]
datasets<6.0.0,>=5.0.1
diffusers>=0.39.0
gymnasium[classic-control]<2.0.0,>=1.2.0
ipykernel>=7.1.0
ipywidgets>=8.1.7
kagglehub>=0.3.0
matplotlib>=3.10.7
numpy>=2.3.4
pandas>=2.2.0
scikit-learn>=1.7.2
seaborn>=0.13.2
torchvision>=0.24.1

[wandb]
wandb>=0.16.0
