FusionFail/requirements.txt at main · atomicarchitects/FusionFail · 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
absl-py==2.1.0
attrs==23.2.0
chex==0.1.86
e3nn-jax==0.20.6
etils==1.8.0
flax==0.8.2
fsspec==2024.3.1
importlib_resources==6.4.0
jax==0.4.25
jaxlib==0.4.25+cuda12.cudnn89
jraph==0.0.6.dev0
markdown-it-py==3.0.0
mdurl==0.1.2
ml-dtypes==0.3.2
mpmath==1.3.0
msgpack==1.0.8
nest-asyncio==1.6.0
numpy==1.26.4
nvidia-cublas-cu12==12.4.2.65
nvidia-cuda-cupti-cu12==12.4.99
nvidia-cuda-nvcc-cu12==12.4.99
nvidia-cuda-runtime-cu12==12.4.99
nvidia-cudnn-cu12==9.0.0.312
nvidia-cufft-cu12==11.2.0.44
nvidia-cusolver-cu12==11.6.0.99
nvidia-cusparse-cu12==12.3.0.142
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.4.99
nvtx==0.2.10
opt-einsum==3.3.0
optax==0.2.2
orbax-checkpoint==0.5.7
protobuf==5.26.0
Pygments==2.17.2
PyYAML==6.0.1
rich==13.7.1
scipy==1.12.0
sympy==1.12
tensorstore==0.1.56
toolz==0.12.1
tqdm==4.66.2
typing_extensions==4.10.0
zipp==3.18.1