JAX_Hypernetworks/requirements.txt at main · Riccardo-Rota/JAX_Hypernetworks · 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
absl-py==2.3.0
chex==0.1.89
etils==1.12.2
flax==0.10.6
fsspec==2025.5.1
humanize==4.12.3
importlib_resources==6.5.2
jax==0.6.1
jax-cuda12-pjrt==0.6.1
jax-cuda12-plugin==0.6.1
jaxlib==0.6.1
markdown-it-py==3.0.0
mdurl==0.1.2
ml_dtypes==0.5.1
msgpack==1.1.0
nest-asyncio==1.6.0
numpy==2.3.0
nvidia-cublas-cu12==12.8.4.1
nvidia-cuda-cupti-cu12==12.9.79
nvidia-cuda-nvcc-cu12==12.9.86
nvidia-cuda-runtime-cu12==12.9.79
nvidia-cudnn-cu12==9.10.2.21
nvidia-cufft-cu12==11.4.1.4
nvidia-cusolver-cu12==11.7.5.82
nvidia-cusparse-cu12==12.5.10.65
nvidia-nccl-cu12==2.27.3
nvidia-nvjitlink-cu12==12.9.86
nvidia-nvshmem-cu12==3.2.5
opt_einsum==3.4.0
optax==0.2.5
orbax-checkpoint==0.11.14
protobuf==6.31.1
Pygments==2.19.1
PyYAML==6.0.2
rich==14.0.0
scipy==1.15.3
setuptools==80.9.0
simplejson==3.20.1
tensorstore==0.1.75
toolz==1.0.0
tqdm==4.67.1
treescope==0.1.9
typing_extensions==4.14.0
zipp==3.23.0