environment.yml

name: jax_env
channels:
  - defaults
dependencies:
  - _libgcc_mutex=0.1=main
  - _openmp_mutex=5.1=1_gnu
  - ca-certificates=2022.4.26=h06a4308_0
  - certifi=2022.6.15=py37h06a4308_0
  - libedit=3.1.20210910=h7f8727e_0
  - libffi=3.2.1=hf484d3e_1007
  - libgcc-ng=11.2.0=h1234567_1
  - libgomp=11.2.0=h1234567_1
  - libstdcxx-ng=11.2.0=h1234567_1
  - ncurses=6.3=h7f8727e_2
  - openssl=1.0.2u=h7b6447c_0
  - pip=21.2.2=py37h06a4308_0
  - python=3.7.0=h6e4f718_3
  - readline=7.0=h7b6447c_5
  - setuptools=61.2.0=py37h06a4308_0
  - sqlite=3.33.0=h62c20be_0
  - tk=8.6.12=h1ccaba5_0
  - wheel=0.37.1=pyhd3eb1b0_0
  - xz=5.2.5=h7f8727e_1
  - zlib=1.2.12=h7f8727e_2
  - pip:
    - absl-py==1.1.0
    - attrs==21.4.0
    - black==22.3.0
    - charset-normalizer==2.1.0
    - chex==0.1.3
    - click==8.1.3
    - codecov==2.1.12
    - coverage==6.4.1
    - cycler==0.11.0
    - dm-tree==0.1.7
    - flake8==4.0.1
    - flatbuffers==2.0
    - flax==0.3.6
    - fonttools==4.33.3
    - idna==3.3
    - importlib-metadata==4.2.0
    - iniconfig==1.1.1
    - isort==5.10.1
    - jax==0.2.26
    - jaxlib==0.3.14
    - kiwisolver==1.4.3
    - matplotlib==3.5.2
    - mccabe==0.6.1
    - msgpack==1.0.4
    - mypy-extensions==0.4.3
    - numpy==1.21.5
    - opt-einsum==3.3.0
    - optax==0.1.0
    - packaging==21.3
    - pathspec==0.9.0
    - pillow==9.1.1
    - platformdirs==2.5.2
    - pluggy==1.0.0
    - py==1.11.0
    - pycodestyle==2.8.0
    - pyflakes==2.4.0
    - pyparsing==3.0.9
    - pytest==6.2.5
    - pytest-cov==3.0.0
    - python-dateutil==2.8.2
    - requests==2.28.1
    - scipy==1.7.3
    - six==1.16.0
    - toml==0.10.2
    - tomli==2.0.1
    - toolz==0.11.2
    - typed-ast==1.5.4
    - typing-extensions==3.10.0.0
    - urllib3==1.26.9
    - zipp==3.8.0