Skip to content

Commit f92267c

Browse files
branch for running dgx spark with requirements.txt
1 parent aa3c664 commit f92267c

1 file changed

Lines changed: 167 additions & 35 deletions

File tree

requirements.txt

Lines changed: 167 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -1,39 +1,171 @@
11
--extra-index-url https://download.pytorch.org/whl/cpu
2-
jax>=0.6.2
3-
jaxlib>=0.4.30
4-
grain
5-
google-cloud-storage>=2.17.0
6-
absl-py
7-
datasets
8-
flax>=0.11.0
9-
optax>=0.2.3
10-
torch>=2.6.0
11-
torchvision>=0.20.1
12-
ftfy
13-
tensorboard>=2.17.0
14-
tensorboardx>=2.6.2.2
15-
tensorboard-plugin-profile>=2.15.2
16-
Jinja2
17-
scikit-image
18-
parameterized
19-
Pillow
20-
pylint
21-
pyink
22-
pytest==8.2.2
23-
tensorflow>=2.17.0
24-
tensorflow-datasets>=4.9.6
25-
ruff>=0.1.5,<=0.2
26-
git+https://github.com/Lightricks/LTX-Video
27-
git+https://github.com/zmelumian972/xla@torchax/jittable_module_callable#subdirectory=torchax
28-
opencv-python-headless==4.10.0.84
29-
orbax-checkpoint
30-
tokenizers==0.21.0
31-
huggingface_hub>=0.30.2
32-
transformers==4.48.1
2+
absl-py==2.3.1
3+
aiofiles==24.1.0
4+
aiohappyeyeballs==2.6.1
5+
aiohttp==3.12.15
6+
aiosignal==1.4.0
7+
annotated-types==0.7.0
8+
aqtp==0.9.0
9+
array_record==0.8.1
10+
astroid==3.3.11
11+
astunparse==1.6.3
12+
attrs==25.3.0
13+
black==24.10.0
14+
cachetools==5.5.2
15+
certifi==2025.8.3
16+
charset-normalizer==3.4.3
17+
cheroot==11.0.0
18+
chex==0.1.91
19+
click==8.3.0
20+
cloudpickle==3.1.1
21+
datasets==4.1.1
22+
decorator==5.2.1
23+
diffusers==0.35.1
24+
dill==0.4.0
25+
dm-tree==0.1.9
26+
docstring_parser==0.17.0
3327
einops==0.8.0
34-
sentencepiece
35-
aqtp
28+
etils==1.13.0
29+
filelock==3.19.1
30+
flatbuffers==25.9.23
31+
flax==0.12.0
32+
frozenlist==1.7.0
33+
fsspec==2025.9.0
34+
ftfy==6.3.1
35+
gast==0.6.0
36+
gcsfs==2025.9.0
37+
google-api-core==2.25.1
38+
google-auth==2.40.3
39+
google-auth-oauthlib==1.2.2
40+
google-cloud-core==2.4.3
41+
google-cloud-storage==3.4.0
42+
google-crc32c==1.7.1
43+
google-pasta==0.2.0
44+
google-resumable-media==2.7.2
45+
googleapis-common-protos==1.70.0
46+
grain==0.2.12
47+
grpcio==1.75.1
48+
gviz-api==1.10.0
49+
h5py==3.14.0
50+
hf-xet==1.1.10
51+
hf_transfer==0.1.9
52+
huggingface-hub==0.35.1
53+
humanize==4.13.0
54+
idna==3.10
3655
imageio==2.37.0
3756
imageio-ffmpeg==0.6.0
38-
hf_transfer>=0.1.9
39-
qwix@git+https://github.com/google/qwix.git
57+
immutabledict==4.2.1
58+
importlib_metadata==8.7.0
59+
importlib_resources==6.5.2
60+
iniconfig==2.1.0
61+
isort==6.0.1
62+
jaraco.functools==4.3.0
63+
jax==0.7.2
64+
jax-cuda13-pjrt==0.7.2
65+
jax-cuda13-plugin==0.7.2
66+
jaxlib==0.7.2
67+
Jinja2==3.1.6
68+
keras==3.11.3
69+
lazy_loader==0.4
70+
libclang==18.1.1
71+
ltx-video @ git+https://github.com/Lightricks/LTX-Video@53d263f31727a0021bf65e3e413d1cb139abb86b
72+
Markdown==3.9
73+
markdown-it-py==4.0.0
74+
MarkupSafe==3.0.2
75+
maxdiffusion @ file:///app
76+
mccabe==0.7.0
77+
mdurl==0.1.2
78+
ml_dtypes==0.5.3
79+
more-itertools==10.8.0
80+
mpmath==1.3.0
81+
msgpack==1.1.1
82+
multidict==6.6.4
83+
multiprocess==0.70.16
84+
mypy_extensions==1.1.0
85+
namex==0.1.0
86+
nest-asyncio==1.6.0
87+
networkx==3.5
88+
numpy==2.3.3
89+
oauthlib==3.3.1
90+
opencv-python-headless==4.10.0.84
91+
opt_einsum==3.4.0
92+
optax==0.2.6
93+
optree==0.17.0
94+
orbax-checkpoint==0.11.25
95+
packaging==25.0
96+
pandas==2.3.2
97+
parameterized==0.9.0
98+
pathspec==0.12.1
99+
pillow==11.3.0
100+
platformdirs==4.4.0
101+
pluggy==1.6.0
102+
promise==2.3
103+
propcache==0.3.2
104+
proto-plus==1.26.1
105+
protobuf==6.32.1
106+
psutil==7.1.0
107+
pyarrow==21.0.0
108+
pyasn1==0.6.1
109+
pyasn1_modules==0.4.2
110+
pybind11==3.0.1
111+
pydantic==2.11.9
112+
pydantic_core==2.33.2
113+
Pygments==2.19.2
114+
pyink==24.10.1
115+
pylint==3.3.8
116+
pytest==8.2.2
117+
python-dateutil==2.9.0.post0
118+
pytz==2025.2
119+
PyYAML==6.0.3
120+
qwix @ git+https://github.com/google/qwix.git@e99e27f517ec8de00c29dd39c638e44e14a215fd
121+
regex==2025.9.18
122+
requests==2.32.5
123+
requests-oauthlib==2.0.0
124+
rich==14.1.0
125+
rsa==4.9.1
126+
ruff==0.2.0
127+
safetensors==0.6.2
128+
scikit-image==0.25.2
129+
scipy==1.16.2
130+
sentencepiece==0.2.1
131+
setuptools==68.1.2
132+
simple-parsing==0.1.7
133+
simplejson==3.20.2
134+
six==1.17.0
135+
sympy==1.14.0
136+
tensorboard==2.20.0
137+
tensorboard-data-server==0.7.2
138+
tensorboard_plugin_profile==2.20.6
139+
tensorboardX==2.6.4
140+
tensorflow==2.20.0
141+
tensorflow-datasets==4.9.9
142+
tensorflow-metadata==1.17.2
143+
tensorstore==0.1.77
144+
termcolor==3.1.0
145+
tifffile==2025.9.20
146+
timm==1.0.20
147+
tokenizers==0.21.0
148+
toml==0.10.2
149+
tomlkit==0.13.3
150+
toolz==1.0.0
151+
torch==2.8.0+cpu
152+
torchax @ git+https://github.com/zmelumian972/xla@f1f6894bb63bfd1d30893f6da500fbfced72b5cd#subdirectory=torchax
153+
torchvision==0.23.0
154+
tqdm==4.67.1
155+
transformer-engine-jax==2.6.0
156+
transformer_engine==2.6.0
157+
transformer_engine_cu12==2.6.0
158+
transformers==4.48.1
159+
treescope==0.1.10
160+
typing-inspection==0.4.1
161+
typing_extensions==4.15.0
162+
tzdata==2025.2
163+
urllib3==2.5.0
164+
wcwidth==0.2.14
165+
Werkzeug==3.1.3
166+
wheel==0.42.0
167+
wrapt==1.17.3
168+
xprof==2.20.6
169+
xxhash==3.5.0
170+
yarl==1.20.1
171+
zipp==3.23.0

0 commit comments

Comments
 (0)