# Python dependencies for source training/evaluation.
# Versions relaxed to floors so the stack installs on modern Python (3.10-3.12)
# and NumPy 2.x, as required by the CUDA 12.8+/PyTorch 2.7+ build used for
# NVIDIA Blackwell GPUs (e.g. RTX 5090, sm_120). Install PyTorch separately
# from the correct CUDA index first (see install.sh).
scikit-learn>=1.3
pickleshare>=0.7.5
ninja>=1.11
gdown
easydict>=1.10
PyYAML>=6.0.1
protobuf>=4.23
tensorboard>=2.14
termcolor>=2.0
tqdm>=4.62
multimethod>=1.9
h5py>=3.9
matplotlib>=3.7
wandb
pyvista
setuptools>=68
Cython>=3.0
pandas
deepspeed
shortuuid

# for docs
mkdocs-material
mkdocs-awesome-pages-plugin
mdx_truly_sane_lists
