requirements_gpu.txt

# install numpy via conda first to receive Intel MKL support

-f https://storage.googleapis.com/jax-releases/jax_releases.html
absl-py==0.12.0
appnope==0.1.2
argon2-cffi==20.1.0
asax @ git+https://github.com/sirmarcel/asax.git@a9ea2dcbd24c9563d55e634bbecb57c280d32ba6
ase @ git+https://gitlab.com/ase/ase.git@a9ea2dcbd24c9563d55e634bbecb57c280d32ba6
async-generator==1.10
attrs==21.2.0
backcall==0.2.0
bleach==3.3.0
cffi==1.14.5
cycler==0.10.0
dataclasses==0.6
decorator==5.0.9
defusedxml==0.7.1
dm-haiku==0.0.4
entrypoints==0.3
flatbuffers==2.0
ipykernel==5.5.5
ipython==7.24.0
ipython-genutils==0.2.0
ipywidgets==7.6.3
jax==0.2.13
jax-md==0.1.15
jaxlib==0.1.67+cuda111
jedi==0.18.0
Jinja2==3.0.1
jsonschema==3.2.0
jupyter==1.0.0
jupyter-client==6.1.12
jupyter-console==6.4.0
jupyter-core==4.7.1
jupyterlab-pygments==0.1.2
jupyterlab-widgets==1.0.0
kiwisolver==1.3.1
MarkupSafe==2.0.1
matplotlib==3.4.2
matplotlib-inline==0.1.2
mistune==0.8.4
nbclient==0.5.3
nbconvert==6.0.7
nbformat==5.1.3
nest-asyncio==1.5.1
notebook==6.4.0
opt-einsum==3.3.0
packaging==20.9
pandas==1.2.4
pandocfilters==1.4.3
parso==0.8.2
pexpect==4.8.0
pickleshare==0.7.5
Pillow==8.2.0
prometheus-client==0.10.1
prompt-toolkit==3.0.18
ptyprocess==0.7.0
pycparser==2.20
Pygments==2.9.0
pyparsing==2.4.7
pyrsistent==0.17.3
python-dateutil==2.8.1
pytz==2021.1
pyzmq==22.1.0
qtconsole==5.1.0
QtPy==1.9.0
scipy==1.6.3
seaborn==0.11.1
Send2Trash==1.5.0
six==1.16.0
tabulate==0.8.9
terminado==0.10.0
testpath==0.5.0
tornado==6.1
traitlets==5.0.5
wcwidth==0.2.5
webencodings==0.5.1
widgetsnbextension==3.5.1