StableTrain/requirements.txt at main · cidl-auth/StableTrain · 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
# Core scientific stack
numpy==2.4.1
scipy==1.17.0
pandas==3.0.0
scikit-learn==1.8.0
matplotlib==3.10.8
seaborn==0.13.2
tqdm==4.67.1

# Imaging / vision
opencv-python==4.13.0.90
pillow==12.1.0
scikit-image==0.26.0
imageio==2.37.2
imgaug==0.4.0
tifffile==2026.1.14
shapely==2.1.2

# Deep learning (PyTorch)
torch==2.4.1
torchvision==0.19.1
torchmetrics==1.8.2
triton==3.0.0
einops==0.8.2
timm==1.0.24
kornia==0.8.2
kornia_rs==0.1.10
safetensors==0.7.0
lightning==2.6.0
lightning-utilities==0.15.2

# Logging / config / CLI
tensorboard==2.20.0
PyYAML==6.0.3
omegaconf==2.3.0
jsonargparse==4.45.0
rich==14.3.1
rich-argparse==1.7.2
python-dotenv==1.2.1

# Web / HTTP utilities (often pulled, but listed explicitly)
requests==2.32.5
urllib3==2.6.3
certifi==2026.1.4
charset-normalizer==3.4.4
idna==3.11

# Transformers ecosystem
huggingface-hub==1.3.4
regex==2026.1.15
filelock==3.20.3
fsspec==2026.1.0

# Visualization extras
plotly==6.5.2
kaleido==1.2.0

# RL / simulation (only keep if this repo actually uses them)
gym==0.26.2
box2d-py==2.3.8
pygame==2.6.1
mujoco==3.4.0
PyOpenGL==3.1.10
glfw==2.10.0
cloudpickle==3.1.2

# Anomaly / research libs (only if used)
anomalib==2.2.0
FrEIA==0.2
open_clip_torch==3.2.0
torchattacks==3.5.1
torch-cka==0.21
simtorch==0.2.1

# TensorFlow stack (ONLY if you truly need TF in this repo)
tensorflow==2.20.0
keras==3.13.1
h5py==3.15.1
absl-py==2.3.1
astunparse==1.6.3
flatbuffers==25.12.19
gast==0.7.0
google-pasta==0.2.0
grpcio==1.76.0
libclang==18.1.1
ml_dtypes==0.5.4
opt-einsum==3.4.0
protobuf==6.33.4
termcolor==3.3.0
wrapt==2.0.1
tensorflow-io-gcs-filesystem==0.37.1

# Misc
psutil==7.2.1
GPUtil==1.4.0
gpustat==1.1.1
pytest==9.0.2
packaging==26.0
typing_extensions==4.15.0
python-dateutil==2.9.0.post0
pytz==2025.2
tzdata==2025.3