numpy
scipy
matplotlib
seaborn
ipykernel
scikit-learn
kagglehub[hf-datasets,pandas-datasets,polars-datasets]
polars
pandas
typer
rich

[cpu]
torch>=2.8
torchvision>=0.23

[cuda]
torch>=2.8
torchvision>=0.23
transformers>=4.57
accelerate>=1.10.1
huggingface_hub>=0.34.4
safetensors>=0.6.2

[xpu]
torch>=2.8
torchvision>=0.23
transformers>=4.57
accelerate>=1.10.1
huggingface_hub>=0.34.4
safetensors>=0.6.2
