bf5fb5f
84f2971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
hydra-core
omegaconf
mlflow
pydantic
scikit-learn
scipy
numpy
pandas
optuna
ray[tune]
transformers
torch
datasets
seaborn
matplotlib
cupy-cuda12x
cuml-cu12==25.2.*
cudf-cu12==25.2.*
torchvision
accelerate |