mfm/environment.yaml at master · albcab/mfm · 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
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
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
name: atmfm
channels:
  - conda-forge
  - defaults
dependencies:
  - _libgcc_mutex=0.1=conda_forge
  - _openmp_mutex=4.5=2_gnu
  - _sysroot_linux-64_curr_repodata_hack=3=h69a702a_14
  - absl-py=2.1.0=py310h06a4308_0
  - aiohttp=3.9.3=py310h5eee18b_0
  - aiosignal=1.2.0=pyhd3eb1b0_0
  - antlr-python-runtime=4.9.3=pyhd8ed1ab_1
  - aom=3.8.2=h59595ed_0
  - appdirs=1.4.4=pyhd3eb1b0_0
  - astunparse=1.6.3=py_0
  - async-timeout=4.0.3=py310h06a4308_0
  - attrs=23.1.0=py310h06a4308_0
  - binutils_impl_linux-64=2.38=h2a08ee3_1
  - binutils_linux-64=2.38.0=hc2dff05_0
  - blackjax=1.0.0=pyhd8ed1ab_0
  - blas=1.0=openblas
  - blinker=1.6.2=py310h06a4308_0
  - blosc=1.21.5=h0f2a231_0
  - bottleneck=1.3.7=py310ha9d4c09_0
  - brotli=1.0.9=h5eee18b_7
  - brotli-bin=1.0.9=h5eee18b_7
  - brotli-python=1.0.9=py310h6a678d5_7
  - bzip2=1.0.8=h5eee18b_5
  - c-ares=1.28.1=hd590300_0
  - ca-certificates=2024.3.11=h06a4308_0
  - cachetools=5.3.3=py310h06a4308_0
  - certifi=2024.2.2=py310h06a4308_0
  - cffi=1.16.0=py310h5eee18b_0
  - charset-normalizer=2.0.4=pyhd3eb1b0_0
  - chex=0.1.86=pyhd8ed1ab_0
  - click=8.1.7=py310h06a4308_0
  - cloudpickle=2.2.1=py310h06a4308_0
  - colorama=0.4.6=py310h06a4308_0
  - contextlib2=21.6.0=py310h06a4308_0
  - contourpy=1.2.0=py310hdb19cb5_0
  - cryptography=42.0.5=py310hdda0065_0
  - cuda-cccl_linux-64=12.4.127=ha770c72_1
  - cuda-crt-dev_linux-64=12.4.131=ha770c72_0
  - cuda-crt-tools=12.4.131=ha770c72_0
  - cuda-cudart=12.4.127=hd3aeb46_0
  - cuda-cudart-dev=12.4.127=hd3aeb46_0
  - cuda-cudart-dev_linux-64=12.4.127=h59595ed_0
  - cuda-cudart-static=12.4.127=hd3aeb46_0
  - cuda-cudart-static_linux-64=12.4.127=h59595ed_0
  - cuda-cudart_linux-64=12.4.127=h59595ed_0
  - cuda-driver-dev_linux-64=12.4.127=h59595ed_0
  - cuda-nvcc=12.4.131=hcdd1206_0
  - cuda-nvcc-dev_linux-64=12.4.131=ha770c72_0
  - cuda-nvcc-impl=12.4.131=hd3aeb46_0
  - cuda-nvcc-tools=12.4.131=hd3aeb46_0
  - cuda-nvcc_linux-64=12.4.131=h8a487aa_0
  - cuda-nvrtc=12.4.127=hd3aeb46_1
  - cuda-nvtx=12.4.127=h59595ed_1
  - cuda-nvvm-dev_linux-64=12.4.131=ha770c72_0
  - cuda-nvvm-impl=12.4.131=h59595ed_0
  - cuda-nvvm-tools=12.4.131=h59595ed_0
  - cuda-version=12.4=h3060b56_3
  - cudnn=8.9.7.29=h092f7fd_3
  - cycler=0.11.0=pyhd3eb1b0_0
  - dataclasses=0.8=pyh6d0b6a4_7
  - dav1d=1.2.1=h5eee18b_0
  - decorator=5.1.1=pyhd3eb1b0_0
  - diffrax=0.5.0=pyhd8ed1ab_0
  - distrax=0.1.5=pyhd8ed1ab_0
  - dm-haiku=0.0.12=pyhd8ed1ab_0
  - dm-tree=0.1.7=py310h6a678d5_1
  - docker-pycreds=0.4.0=pyhd3eb1b0_0
  - equinox=0.11.4=pyhd8ed1ab_0
  - etils=1.7.0=pyhd8ed1ab_0
  - fastprogress=1.0.0=pyhb85f177_0
  - flatbuffers=23.5.26=h59595ed_1
  - flax=0.8.2=pyhd8ed1ab_0
  - fonttools=4.51.0=py310h5eee18b_0
  - freetype=2.12.1=h4a9f257_0
  - frozenlist=1.4.0=py310h5eee18b_0
  - gast=0.5.3=pyhd3eb1b0_0
  - gcc_impl_linux-64=11.2.0=h1234567_1
  - gcc_linux-64=11.2.0=h5c386dc_0
  - giflib=5.2.1=h5eee18b_3
  - gitdb=4.0.7=pyhd3eb1b0_0
  - gitpython=3.1.37=py310h06a4308_0
  - google-auth=2.6.0=pyhd3eb1b0_0
  - google-auth-oauthlib=0.5.2=py310h06a4308_0
  - google-pasta=0.2.0=pyhd3eb1b0_0
  - grpcio=1.59.3=py310h1b8f574_0
  - gxx_impl_linux-64=11.2.0=h1234567_1
  - gxx_linux-64=11.2.0=hc2dff05_0
  - h5py=3.9.0=py310he06866b_0
  - hdf5=1.12.1=h2b7332f_3
  - hydra-core=1.3.2=pyhd8ed1ab_0
  - icu=73.2=h59595ed_0
  - idna=3.4=py310h06a4308_0
  - importlib-metadata=7.1.0=pyha770c72_0
  - importlib_metadata=7.1.0=hd8ed1ab_0
  - importlib_resources=6.1.1=py310h06a4308_1
  - jax=0.4.26=pyhd8ed1ab_0
  - jaxlib=0.4.23=py310h6a678d5_0
  - jaxopt=0.8.3=pyhd8ed1ab_0
  - jaxtyping=0.2.28=pyhd8ed1ab_0
  - jmp=0.0.4=pyhd8ed1ab_0
  - keras=2.15.0=pyhd8ed1ab_0
  - kernel-headers_linux-64=3.10.0=h57e8cba_10
  - kiwisolver=1.4.4=py310h6a678d5_0
  - krb5=1.20.1=h143b758_1
  - lcms2=2.16=hb7c19ff_0
  - ld_impl_linux-64=2.38=h1181459_1
  - lerc=4.0.0=h27087fc_0
  - libabseil=20230802.1=cxx17_h59595ed_0
  - libavif16=1.0.4=hd9d6309_2
  - libblas=3.9.0=22_linux64_openblas
  - libbrotlicommon=1.0.9=h5eee18b_7
  - libbrotlidec=1.0.9=h5eee18b_7
  - libbrotlienc=1.0.9=h5eee18b_7
  - libcblas=3.9.0=22_linux64_openblas
  - libcublas=12.4.5.8=hd3aeb46_1
  - libcufft=11.2.1.3=hd3aeb46_1
  - libcurand=10.3.5.147=hd3aeb46_1
  - libcurl=8.5.0=h251f7ec_0
  - libcusolver=11.6.1.9=hd3aeb46_1
  - libcusparse=12.3.1.170=hd3aeb46_1
  - libdeflate=1.20=hd590300_0
  - libedit=3.1.20230828=h5eee18b_0
  - libev=4.33=h7f8727e_1
  - libffi=3.4.4=h6a678d5_0
  - libgcc-devel_linux-64=11.2.0=h1234567_1
  - libgcc-ng=13.2.0=hc881cc4_6
  - libgfortran-ng=13.2.0=h69a702a_6
  - libgfortran5=13.2.0=h43f5ff8_6
  - libgomp=13.2.0=hc881cc4_6
  - libgrpc=1.59.3=hd6c4280_0
  - libjpeg-turbo=3.0.0=hd590300_1
  - liblapack=3.9.0=22_linux64_openblas
  - libnghttp2=1.57.0=h2d74bed_0
  - libnvjitlink=12.4.127=hd3aeb46_1
  - libopenblas=0.3.27=pthreads_h413a1c8_0
  - libpng=1.6.39=h5eee18b_0
  - libprotobuf=4.24.4=hf27288f_0
  - libre2-11=2023.09.01=h7a70373_1
  - libsqlite=3.45.3=h2797004_0
  - libssh2=1.10.0=hdbd6064_2
  - libstdcxx-devel_linux-64=11.2.0=h1234567_1
  - libstdcxx-ng=13.2.0=h95c4c6d_6
  - libtiff=4.6.0=h1dd3fc0_3
  - libuuid=1.41.5=h5eee18b_0
  - libwebp-base=1.3.2=h5eee18b_0
  - libxcb=1.15=h7f8727e_0
  - libzlib=1.2.13=hd590300_5
  - lineax=0.0.5=pyhd8ed1ab_0
  - lz4-c=1.9.4=h6a678d5_0
  - markdown=3.4.1=py310h06a4308_0
  - markdown-it-py=2.2.0=py310h06a4308_1
  - markupsafe=2.1.3=py310h5eee18b_0
  - matplotlib-base=3.8.4=py310h1128e8f_0
  - mdurl=0.1.0=py310h06a4308_0
  - ml-collections=0.1.1=pyhd8ed1ab_0
  - ml_dtypes=0.2.0=py310h1128e8f_0
  - msgpack-python=1.0.3=py310hd09550d_0
  - multidict=6.0.4=py310h5eee18b_0
  - multipledispatch=0.6.0=py310h06a4308_0
  - nccl=2.21.5.1=h3a97aeb_0
  - ncurses=6.4=h6a678d5_0
  - nest-asyncio=1.6.0=py310h06a4308_0
  - numexpr=2.8.7=py310h286c3b5_0
  - numpy=1.26.4=py310hb13e2d6_0
  - numpyro=0.14.0=pyhd8ed1ab_0
  - oauthlib=3.2.2=py310h06a4308_0
  - omegaconf=2.3.0=pyhd8ed1ab_0
  - openjpeg=2.5.0=h488ebb8_3
  - openssl=3.3.0=hd590300_0
  - opt-einsum=3.3.0=hd8ed1ab_2
  - opt_einsum=3.3.0=pyhc1e730c_2
  - optax=0.1.9=pyhd8ed1ab_0
  - optimistix=0.0.6=pyhd8ed1ab_0
  - orbax-checkpoint=0.4.4=pyhd8ed1ab_0
  - packaging=23.2=py310h06a4308_0
  - pandas=2.2.1=py310h6a678d5_0
  - pathtools=0.1.2=pyhd3eb1b0_1
  - pillow=10.2.0=py310h01dd4db_0
  - pip=23.3.1=py310h06a4308_0
  - psutil=5.9.0=py310h5eee18b_0
  - pyasn1=0.4.8=pyhd3eb1b0_0
  - pyasn1-modules=0.2.8=py_0
  - pybind11-abi=4=hd3eb1b0_1
  - pycparser=2.21=pyhd3eb1b0_0
  - pygments=2.15.1=py310h06a4308_1
  - pyjwt=2.4.0=py310h06a4308_0
  - pyopenssl=24.0.0=py310h06a4308_0
  - pyparsing=3.0.9=py310h06a4308_0
  - pysocks=1.7.1=py310h06a4308_0
  - python=3.10.14=h955ad1f_0
  - python-dateutil=2.8.2=pyhd3eb1b0_0
  - python-flatbuffers=24.3.25=pyh59ac667_0
  - python-tzdata=2023.3=pyhd3eb1b0_0
  - python_abi=3.10=2_cp310
  - pytz=2024.1=py310h06a4308_0
  - pyyaml=6.0.1=py310h5eee18b_0
  - rav1e=0.6.6=he8a937b_2
  - re2=2023.09.01=h7f4b329_1
  - readline=8.2=h5eee18b_0
  - requests=2.31.0=py310h06a4308_1
  - requests-oauthlib=1.3.0=py_0
  - rich=13.3.5=py310h06a4308_0
  - rsa=4.7.2=pyhd3eb1b0_1
  - scipy=1.13.0=py310hb13e2d6_0
  - sentry-sdk=1.9.0=py310h06a4308_0
  - setproctitle=1.2.2=py310h7f8727e_0
  - setuptools=68.2.2=py310h06a4308_0
  - six=1.16.0=pyhd3eb1b0_1
  - smmap=4.0.0=pyhd3eb1b0_0
  - snappy=1.1.10=h6a678d5_1
  - sqlite=3.41.2=h5eee18b_0
  - svt-av1=2.0.0=h59595ed_0
  - sysroot_linux-64=2.17=h57e8cba_10
  - tabulate=0.9.0=py310h06a4308_0
  - tensorboard=2.15.2=pyhd8ed1ab_0
  - tensorboard-data-server=0.7.0=py310h52d8a92_0
  - tensorflow-base=2.15.0=cpu_py310h7e4d085_2
  - tensorflow-probability=0.23.0=pyh4af843d_0
  - termcolor=2.1.0=py310h06a4308_0
  - tk=8.6.13=noxft_h4845f30_101
  - toolz=0.12.0=py310h06a4308_0
  - tqdm=4.66.2=pyhd8ed1ab_0
  - typeguard=2.13.3=py310h06a4308_0
  - typing-extensions=4.9.0=py310h06a4308_1
  - typing_extensions=4.9.0=py310h06a4308_1
  - tzdata=2024a=h04d1e81_0
  - unicodedata2=15.1.0=py310h5eee18b_0
  - urllib3=2.1.0=py310h06a4308_1
  - wandb=0.16.5=pyhd8ed1ab_0
  - werkzeug=2.3.8=py310h06a4308_0
  - wheel=0.41.2=py310h06a4308_0
  - wrapt=1.14.1=py310h5eee18b_0
  - xz=5.4.6=h5eee18b_0
  - yaml=0.2.5=h7b6447c_0
  - yarl=1.9.3=py310h5eee18b_0
  - zipp=3.17.0=pyhd8ed1ab_0
  - zlib=1.2.13=hd590300_5
  - zstd=1.5.5=hc292b87_0
  - pip:
      - annealed-flow-transport==1.0
      - array-record==0.5.1
      - exceptiongroup==1.2.1
      - fsspec==2024.3.1
      - inference-gym==0.0.4
      - iniconfig==2.0.0
      - jaxline==0.0.8
      - pluggy==1.5.0
      - promise==2.3
      - protobuf==3.20.3
      - pytest==8.2.0
      - tensorflow-datasets==4.9.4
      - tensorflow-estimator==2.15.0
      - tensorflow-metadata==1.15.0
      - tensorstore==0.1.45
      - tfp-nightly==0.25.0.dev20240430
      - toml==0.10.2
      - tomli==2.0.1
prefix: /storage/hpc/14/cabezasg/.conda/envs/atmfm