#
# This file is autogenerated by pip-compile with python 3.8
# To update, run:
#
#    pip-compile requirements.in
#
--find-links https://storage.googleapis.com/jax-releases/jax_releases.html

absl-py==1.0.0
    # via
    #   chex
    #   jax
    #   jaxlib
    #   optax
chex==0.1.1
    # via
    #   -r requirements.in
    #   optax
dm-tree==0.1.6
    # via chex
flatbuffers==2.0
    # via jaxlib
jax[cuda]==0.3.1
    # via
    #   -r requirements.in
    #   chex
    #   optax
jaxlib==0.3.0+cuda11.cudnn82
    # via
    #   chex
    #   jax
    #   optax
numpy==1.22.2
    # via
    #   chex
    #   jax
    #   jaxlib
    #   opt-einsum
    #   optax
    #   scipy
opt-einsum==3.3.0
    # via jax
optax==0.1.1
    # via -r requirements.in
scipy==1.8.0
    # via
    #   jax
    #   jaxlib
six==1.16.0
    # via
    #   absl-py
    #   dm-tree
toolz==0.11.2
    # via chex
typing-extensions==4.1.1
    # via
    #   jax
    #   optax
