MLEvolve/requirements_ml.txt at main · InternScience/MLEvolve · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
nvidia-cublas-cu12==12.6.4.1
nvidia-cuda-cupti-cu12==12.6.80
nvidia-cuda-nvrtc-cu12==12.6.77
nvidia-cuda-runtime-cu12==12.6.77
nvidia-cudnn-cu12==9.5.1.17
nvidia-cufft-cu12==11.3.0.4
nvidia-cufile-cu12==1.11.1.6
nvidia-curand-cu12==10.3.7.77
nvidia-cusolver-cu12==11.7.1.2
nvidia-cusparse-cu12==12.5.4.2
nvidia-cusparselt-cu12==0.6.3
nvidia-ml-py3==7.352.0
nvidia-nccl-cu12==2.26.2
nvidia-nvjitlink-cu12==12.6.85
nvidia-nvshmem-cu12==3.3.20
nvidia-nvtx-cu12==12.6.77


torch==2.7.1
torchvision==0.22.1
torchaudio==2.7.1
triton==3.3.1


accelerate==1.11.0
lightning==2.5.2
lightning-utilities==0.14.1
pytorch-lightning==2.5.1
torchmetrics==1.0.0
torchinfo==1.8.0
torchdata==0.11.0
torchtext==0.18.0
einops==0.8.1
einx==0.3.0
timm==1.0.22
peft==0.14.0
optimum==1.26.1
optree==0.14.1


huggingface-hub==0.36.0
hf-xet==1.2.0
transformers==4.57.1
tokenizers==0.22.1
datasets==4.1.1
evaluate==0.4.5
sentencepiece==0.2.1
sentence-transformers==2.6.1


tensorflow==2.19.0
tensorflow-hub==0.16.1
tensorflow-text==2.19.0
tf_keras==2.19.0
keras==3.9.0
libclang==18.1.1
tensorboard==2.19.0
tensorboard-data-server==0.7.2
tensorboardX==2.6.4
tensorpack==0.11


scikit-learn==1.6.1
scikit-image==0.25.2
scikit-base==0.13.0
scikit-multilearn==0.2.0
scikit-optimize==0.10.2
sklearn-compat==0.1.3
sklearn-pandas==2.2.0
imbalanced-learn==0.13.0
category_encoders==2.8.1
shap==0.45.1

lightgbm==4.6.0
xgboost==3.1.1
catboost==1.2.8
pytorch-tabnet==4.1.0
hyperopt==0.2.7
bayesian-optimization==2.0.3


autogluon==1.4.0
autogluon.common==1.4.0
autogluon.core==1.4.0
autogluon.features==1.4.0
autogluon.multimodal==1.4.0
autogluon.tabular==1.4.0
autogluon.timeseries==1.4.0
optuna==4.2.1
optuna-integration==4.3.0
ray==2.44.1


onnx==1.18.0
onnxruntime-gpu==1.22.0


statsmodels==0.14.4
numba==0.61.0
llvmlite==0.44.0


black==24.3.0
confection==0.1.5
catalogue==2.0.10
beartype==0.20.0
plum-dispatch==2.5.7