environment-jax.yml

name: /scratch/gpfs/altosaar/environment-jax
channels:
  - defaults
dependencies:
  - _libgcc_mutex=0.1=main
  - ca-certificates=2021.4.13=h06a4308_1
  - certifi=2020.12.5=py39h06a4308_0
  - ld_impl_linux-64=2.33.1=h53a641e_7
  - libffi=3.3=he6710b0_2
  - libgcc-ng=9.1.0=hdf63c60_0
  - libstdcxx-ng=9.1.0=hdf63c60_0
  - ncurses=6.2=he6710b0_1
  - openssl=1.1.1k=h27cfd23_0
  - python=3.9.5=hdb3f193_3
  - readline=8.1=h27cfd23_0
  - setuptools=52.0.0=py39h06a4308_0
  - sqlite=3.35.4=hdfb4753_0
  - tk=8.6.10=hbc83047_0
  - tzdata=2020f=h52ac0ba_0
  - wheel=0.36.2=pyhd3eb1b0_0
  - xz=5.2.5=h7b6447c_0
  - zlib=1.2.11=h7b6447c_3
  - pip:
    - absl-py==0.12.0
    - astunparse==1.6.3
    - attrs==21.2.0
    - cachetools==4.2.2
    - chardet==4.0.0
    - chex==0.0.7
    - cloudpickle==1.6.0
    - decorator==5.0.9
    - dill==0.3.3
    - dm-haiku==0.0.5.dev0
    - dm-tree==0.1.6
    - flatbuffers==1.12
    - future==0.18.2
    - gast==0.4.0
    - google-auth==1.30.1
    - google-auth-oauthlib==0.4.4
    - google-pasta==0.2.0
    - googleapis-common-protos==1.53.0
    - grpcio==1.38.0
    - h5py==3.1.0
    - idna==2.10
    - jax==0.2.13
    - jaxlib==0.1.67+cuda111
    - jmp==0.0.2
    - keras-nightly==2.6.0.dev2021052500
    - keras-preprocessing==1.1.2
    - markdown==3.3.4
    - numpy==1.19.5
    - oauthlib==3.1.0
    - opt-einsum==3.3.0
    - optax==0.0.6
    - pip==21.1.2
    - promise==2.3
    - protobuf==3.17.1
    - pyasn1==0.4.8
    - pyasn1-modules==0.2.8
    - requests==2.25.1
    - requests-oauthlib==1.3.0
    - rsa==4.7.2
    - scipy==1.6.3
    - six==1.15.0
    - tabulate==0.8.9
    - tb-nightly==2.6.0a20210525
    - tensorboard-data-server==0.6.1
    - tensorboard-plugin-wit==1.8.0
    - tensorflow-datasets==4.3.0
    - tensorflow-metadata==1.0.0
    - termcolor==1.1.0
    - tf-estimator-nightly==2.5.0.dev2021032601
    - tf-nightly==2.6.0.dev20210525
    - tfp-nightly==0.14.0.dev20210525
    - toolz==0.11.1
    - tqdm==4.61.0
    - typing-extensions==3.7.4.3
    - urllib3==1.26.4
    - werkzeug==2.0.1
    - wrapt==1.12.1
prefix: /scratch/gpfs/altosaar/environment-jax