-
Notifications
You must be signed in to change notification settings - Fork 0
/
jax_env.yml
176 lines (176 loc) · 5 KB
/
jax_env.yml
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
name: jax_env
channels:
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- archspec=0.2.1=pyhd3eb1b0_0
- boltons=23.0.0=py310h06a4308_0
- brotli-python=1.0.9=py310h6a678d5_8
- bzip2=1.0.8=h7b6447c_0
- c-ares=1.19.1=h5eee18b_0
- ca-certificates=2024.3.11=h06a4308_0
- certifi=2024.2.2=py310h06a4308_0
- cffi=1.16.0=py310h5eee18b_1
- charset-normalizer=2.0.4=pyhd3eb1b0_0
- conda-content-trust=0.2.0=py310h06a4308_1
- conda-package-handling=2.2.0=py310h06a4308_1
- conda-package-streaming=0.9.0=py310h06a4308_0
- cryptography=41.0.7=py310hdda0065_0
- distro=1.8.0=py310h06a4308_0
- expat=2.5.0=h6a678d5_0
- fmt=9.1.0=hdb19cb5_0
- icu=73.1=h6a678d5_0
- idna=3.4=py310h06a4308_0
- jsonpatch=1.32=pyhd3eb1b0_0
- jsonpointer=2.1=pyhd3eb1b0_0
- krb5=1.20.1=h143b758_1
- ld_impl_linux-64=2.38=h1181459_1
- libarchive=3.6.2=h6ac8c49_2
- libcurl=8.5.0=h251f7ec_0
- libedit=3.1.20230828=h5eee18b_0
- libev=4.33=h7f8727e_1
- libffi=3.4.4=h6a678d5_0
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libmamba=1.5.3=haf1ee3a_0
- libmambapy=1.5.3=py310h2dafd23_0
- libnghttp2=1.57.0=h2d74bed_0
- libsolv=0.7.24=he621ea3_0
- libssh2=1.10.0=hdbd6064_2
- libstdcxx-ng=11.2.0=h1234567_1
- libuuid=1.41.5=h5eee18b_0
- libxml2=2.10.4=hf1b16e4_1
- lz4-c=1.9.4=h6a678d5_0
- menuinst=2.0.2=py310h06a4308_1
- ncurses=6.4=h6a678d5_0
- openssl=3.0.13=h7f8727e_1
- packaging=23.1=py310h06a4308_0
- pcre2=10.42=hebb0a14_0
- platformdirs=3.10.0=py310h06a4308_0
- pluggy=1.0.0=py310h06a4308_1
- pybind11-abi=4=hd3eb1b0_1
- pycosat=0.6.6=py310h5eee18b_1
- pycparser=2.21=pyhd3eb1b0_0
- pysocks=1.7.1=py310h06a4308_0
- python=3.10.13=h955ad1f_0
- readline=8.2=h5eee18b_0
- reproc=14.2.4=h295c915_1
- reproc-cpp=14.2.4=h295c915_1
- requests=2.31.0=py310h06a4308_1
- ruamel.yaml=0.17.21=py310h5eee18b_0
- ruamel.yaml.clib=0.2.6=py310h5eee18b_1
- setuptools=68.2.2=py310h06a4308_0
- sqlite=3.41.2=h5eee18b_0
- tk=8.6.12=h1ccaba5_0
- tqdm=4.65.0=py310h2f386ee_0
- truststore=0.8.0=py310h06a4308_0
- tzdata=2023d=h04d1e81_0
- urllib3=2.1.0=py310h06a4308_1
- wheel=0.41.2=py310h06a4308_0
- xz=5.4.5=h5eee18b_0
- yaml-cpp=0.8.0=h6a678d5_0
- zlib=1.2.13=h5eee18b_0
- zstandard=0.19.0=py310h5eee18b_0
- zstd=1.5.5=hc292b87_0
- pip:
- absl-py==2.1.0
- array-record==0.5.1
- astunparse==1.6.3
- cachetools==5.3.3
- chex==0.1.86
- click==8.1.7
- diffrax==0.5.1
- dm-tree==0.1.8
- equinox==0.11.4
- etils==1.7.0
- filelock==3.14.0
- flatbuffers==24.3.25
- flax==0.8.3
- fsspec==2024.5.0
- gast==0.5.4
- google-auth==2.29.0
- google-auth-oauthlib==1.2.0
- google-pasta==0.2.0
- grpcio==1.64.0
- h5py==3.11.0
- importlib-resources==6.4.0
- jax==0.4.28
- jax-cuda12-pjrt==0.4.28
- jax-cuda12-plugin==0.4.28
- jaxlib==0.4.28
- jaxtyping==0.2.28
- jinja2==3.1.4
- keras==2.15.0
- libclang==18.1.1
- lineax==0.0.5
- markdown==3.6
- markdown-it-py==3.0.0
- markupsafe==2.1.5
- mdurl==0.1.2
- ml-dtypes==0.2.0
- mpmath==1.3.0
- msgpack==1.0.8
- namex==0.0.8
- nest-asyncio==1.6.0
- networkx==3.3
- numpy==1.26.4
- nvidia-cublas-cu12==12.1.3.1
- nvidia-cuda-cupti-cu12==12.1.105
- nvidia-cuda-nvcc-cu12==12.2.140
- nvidia-cuda-nvrtc-cu12==12.1.105
- nvidia-cuda-runtime-cu12==12.1.105
- nvidia-cudnn-cu12==8.9.2.26
- nvidia-cufft-cu12==11.0.2.54
- nvidia-curand-cu12==10.3.2.106
- nvidia-cusolver-cu12==11.4.5.107
- nvidia-cusparse-cu12==12.1.0.106
- nvidia-nccl-cu12==2.20.5
- nvidia-nvjitlink-cu12==12.2.140
- nvidia-nvtx-cu12==12.1.105
- oauthlib==3.2.2
- opt-einsum==3.3.0
- optax==0.2.2
- optimistix==0.0.7
- optree==0.11.0
- orbax==0.1.9
- orbax-checkpoint==0.5.11
- pillow==10.3.0
- pip==24.0
- promise==2.3
- protobuf==3.20.3
- psutil==5.9.8
- pyasn1==0.6.0
- pyasn1-modules==0.4.0
- pygments==2.18.0
- pyyaml==6.0.1
- requests-oauthlib==2.0.0
- rich==13.7.1
- rsa==4.9
- scipy==1.13.0
- six==1.16.0
- sympy==1.12
- tensorboard==2.15.2
- tensorboard-data-server==0.7.2
- tensorflow-datasets==4.9.4
- tensorflow-estimator==2.15.0
- tensorflow-io-gcs-filesystem==0.37.0
- tensorflow-metadata==1.15.0
- tensorrt==10.0.1
- tensorrt-cu12==10.0.1
- tensorrt-cu12-bindings==10.0.1
- tensorrt-cu12-libs==10.0.1
- tensorstore==0.1.59
- termcolor==2.4.0
- toml==0.10.2
- toolz==0.12.1
- torch==2.3.0
- torchaudio==2.3.0
- torchvision==0.18.0
- triton==2.3.0
- typeguard==2.13.3
- typing-extensions==4.11.0
- werkzeug==3.0.3
- wrapt==1.14.1
- zipp==3.18.2
prefix: /home/annabel/miniconda3/envs/jax_env