#

# This file is autogenerated by pip-compile with Python 3.10

# by the following command:

#

# pip-compile --allow-unsafe --output-file=requirements-cuda.txt requirements-cuda.in

#

--find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html

--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

absl-py==1.4.0

# via

# -r requirements.in

# array-record

# chex

# clu

# fiddle

# mesh-tensorflow

# ml-collections

# optax

# orbax-checkpoint

# orbax-export

# paxml

# praxis

# rouge-score

# seqio

# seqio-nightly

# t5

# tensorboard

# tensorflow

# tensorflow-datasets

# tensorflow-metadata

# tfds-nightly

anyio==3.7.1

# via jupyter-server

argon2-cffi==23.1.0

# via jupyter-server

argon2-cffi-bindings==21.2.0

# via argon2-cffi

array-record==0.4.1

# via tfds-nightly

arrow==1.2.3

# via isoduration

asttokens==2.2.1

# via stack-data

astunparse==1.6.3

# via tensorflow

async-lru==2.0.4

# via jupyterlab

attrs==23.1.0

# via

# jsonschema

# lingvo

# referencing

babel==2.12.1

# via

# jupyterlab-server

# t5

backcall==0.2.0

# via ipython

beautifulsoup4==4.12.2

# via nbconvert

bleach==6.0.0

# via nbconvert

cachetools==5.3.1

# via google-auth

certifi==2023.7.22

# via requests

cffi==1.15.1

# via argon2-cffi-bindings

charset-normalizer==3.2.0

# via requests

chex==0.1.82

# via optax

click==8.1.7

# via

# nltk

# tensorflow-datasets

# tfds-nightly

clu==0.0.9

# via

# paxml

# praxis

# seqio

# seqio-nightly

colorama==0.4.6

# via sacrebleu

comm==0.1.4

# via

# ipykernel

# ipywidgets

contextlib2==21.6.0

# via ml-collections

contourpy==1.1.0

# via matplotlib

cycler==0.11.0

# via matplotlib

dataclasses-json==0.5.14

# via orbax-export

debugpy==1.6.7.post1

# via ipykernel

decorator==5.1.1

# via ipython

defusedxml==0.7.1

# via nbconvert

dm-tree==0.1.8

# via

# tensorflow-datasets

# tfds-nightly

docstring-parser==0.15

# via pyglove

editdistance==0.6.2

# via

# seqio

# seqio-nightly

# t5

einops==0.6.1

# via praxis

etils[enp,epath,epy]==1.4.1

# via

# array-record

# clu

# lingvo

# orbax-checkpoint

# orbax-export

# paxml

# praxis

# tensorflow-datasets

# tfds-nightly

exceptiongroup==1.1.3

# via anyio

executing==1.2.0

# via stack-data

fastjsonschema==2.18.0

# via nbformat

fiddle @ git+https://github.com/google/fiddle

# via praxis

filelock==3.12.2

# via

# huggingface-hub

# transformers

flatbuffers==1.12

# via tensorflow

flax==0.7.2

# via

# clu

# paxml

# praxis

fonttools==4.42.1

# via matplotlib

fqdn==1.5.1

# via jsonschema

fsspec==2023.6.0

# via huggingface-hub

future==0.18.3

# via mesh-tensorflow

gast==0.4.0

# via tensorflow

gin-config==0.5.0

# via

# mesh-tensorflow

# t5

google-auth==2.22.0

# via

# google-auth-oauthlib

# tensorboard

google-auth-oauthlib==0.4.6

# via tensorboard

google-pasta==0.2.0

# via tensorflow

googleapis-common-protos==1.60.0

# via tensorflow-metadata

graph-compression-google-research==0.0.4

# via lingvo

graphviz==0.20.1

# via

# fiddle

# paxml

grpcio==1.57.0

# via

# -r requirements.in

# grpcio-reflection

# tensorboard

# tensorflow

grpcio-reflection==1.48.2

# via -r requirements.in

h5py==3.9.0

# via tensorflow

huggingface-hub==0.16.4

# via transformers

idna==3.4

# via

# anyio

# jsonschema

# requests

immutabledict==3.0.0

# via t5

importlib-metadata==6.8.0

# via -r requirements.in

importlib-resources==6.0.1

# via etils

ipykernel==6.25.1

# via

# jupyter

# jupyter-console

# jupyterlab

# lingvo

# qtconsole

ipython==8.14.0

# via

# ipykernel

# ipywidgets

# jupyter-console

ipython-genutils==0.2.0

# via qtconsole

ipywidgets==8.1.0

# via jupyter

isoduration==20.11.0

# via jsonschema

jax[cuda,tpu] @ git+https://github.com/google/jax

# via

# -r requirements-cuda.in

# -r requirements.in

# chex

# clu

# flax

# optax

# orbax-checkpoint

# orbax-export

# paxml

# praxis

# seqio

# seqio-nightly

jax-bitempered-loss==0.0.2

# via praxis

jaxlib==0.4.14+cuda11.cudnn86

# via

# chex

# clu

# jax

# optax

# orbax-checkpoint

# orbax-export

# seqio

# seqio-nightly

jaxtyping==0.2.21

# via praxis

jedi==0.19.0

# via ipython

jinja2==3.1.2

# via

# jupyter-server

# jupyterlab

# jupyterlab-server

# nbconvert

joblib==1.3.2

# via

# nltk

# scikit-learn

json5==0.9.14

# via jupyterlab-server

jsonpointer==2.4

# via jsonschema

jsonschema[format-nongpl]==4.19.0

# via

# jupyter-events

# jupyterlab-server

# nbformat

jsonschema-specifications==2023.7.1

# via jsonschema

jupyter==1.0.0

# via lingvo

jupyter-client==8.3.0

# via

# ipykernel

# jupyter-console

# jupyter-server

# nbclient

# qtconsole

jupyter-console==6.6.3

# via jupyter

jupyter-core==5.3.1

# via

# ipykernel

# jupyter-client

# jupyter-console

# jupyter-server

# jupyterlab

# nbclient

# nbconvert

# nbformat

# qtconsole

jupyter-events==0.7.0

# via jupyter-server

jupyter-http-over-ws==0.0.8

# via lingvo

jupyter-lsp==2.2.0

# via jupyterlab

jupyter-server==2.7.2

# via

# jupyter-lsp

# jupyterlab

# jupyterlab-server

# notebook

# notebook-shim

jupyter-server-terminals==0.4.4

# via jupyter-server

jupyterlab==4.0.5

# via notebook

jupyterlab-pygments==0.2.2

# via nbconvert

jupyterlab-server==2.24.0

# via

# jupyterlab

# notebook

jupyterlab-widgets==3.0.8

# via ipywidgets

keras==2.9.0

# via tensorflow

keras-preprocessing==1.1.2

# via tensorflow

kiwisolver==1.4.5

# via matplotlib

libclang==16.0.6

# via tensorflow

libcst==1.0.1

# via fiddle

libtpu-nightly==0.1.dev20230727

# via jax

lingvo==0.12.7

# via

# paxml

# praxis

lxml==4.9.3

# via sacrebleu

markdown==3.4.4

# via tensorboard

markdown-it-py==3.0.0

# via rich

markupsafe==2.1.3

# via

# jinja2

# nbconvert

# werkzeug

marshmallow==3.20.1

# via dataclasses-json

matplotlib==3.7.2

# via lingvo

matplotlib-inline==0.1.6

# via

# ipykernel

# ipython

mdurl==0.1.2

# via markdown-it-py

mesh-tensorflow[transformer]==0.1.21

# via t5

mistune==3.0.1

# via nbconvert

ml-collections==0.1.1

# via clu

ml-dtypes==0.2.0

# via

# jax

# jaxlib

model-pruning-google-research==0.0.5

# via lingvo

mpmath==1.3.0

# via sympy

msgpack==1.0.5

# via

# flax

# orbax-checkpoint

mypy-extensions==1.0.0

# via typing-inspect

nbclient==0.8.0

# via nbconvert

nbconvert==7.7.4

# via

# jupyter

# jupyter-server

nbformat==5.9.2

# via

# jupyter-server

# nbclient

# nbconvert

nest-asyncio==1.5.7

# via

# ipykernel

# orbax-checkpoint

nltk==3.8.1

# via

# rouge-score

# t5

notebook==7.0.2

# via

# jupyter

# jupyter-http-over-ws

notebook-shim==0.2.3

# via

# jupyterlab

# notebook

numpy==1.25.2

# via

# -r requirements.in

# chex

# clu

# contourpy

# etils

# flax

# h5py

# jax

# jaxlib

# jaxtyping

# keras-preprocessing

# matplotlib

# ml-dtypes

# opt-einsum

# optax

# orbax-checkpoint

# orbax-export

# pandas

# paxml

# praxis

# rouge-score

# sacrebleu

# scikit-learn

# scipy

# seqio

# seqio-nightly

# t5

# tensorboard

# tensorflow

# tensorflow-datasets

# tensorflow-hub

# tensorstore

# tfds-nightly

# transformers

oauthlib==3.2.2

# via requests-oauthlib

opt-einsum==3.3.0

# via

# jax

# tensorflow

optax==0.1.7

# via

# flax

# praxis

optax-shampoo==0.0.6

# via praxis

orbax-checkpoint==0.3.5

# via

# flax

# orbax-export

# paxml

orbax-export==0.0.3

# via -r requirements.in

overrides==7.4.0

# via jupyter-server

packaging==23.1

# via

# clu

# huggingface-hub

# ipykernel

# jupyter-server

# jupyterlab

# jupyterlab-server

# marshmallow

# matplotlib

# nbconvert

# qtconsole

# qtpy

# seqio

# seqio-nightly

# tensorflow

# transformers

pandas==2.0.3

# via t5

pandocfilters==1.5.0

# via nbconvert

parso==0.8.3

# via jedi

paxml @ git+https://github.com/google/paxml

# via -r requirements.in

pexpect==4.8.0

# via ipython

pickleshare==0.7.5

# via ipython

pillow==10.0.0

# via

# lingvo

# matplotlib

platformdirs==3.10.0

# via jupyter-core

portalocker==2.7.0

# via sacrebleu

portpicker==1.6.0

# via -r requirements.in

praxis @ git+https://github.com/google/praxis

# via

# -r requirements.in

# paxml

prometheus-client==0.17.1

# via jupyter-server

promise==2.3

# via

# tensorflow-datasets

# tfds-nightly

prompt-toolkit==3.0.39

# via

# ipython

# jupyter-console

protobuf==3.20.3

# via

# googleapis-common-protos

# grpcio-reflection

# lingvo

# orbax-checkpoint

# paxml

# seqio-nightly

# tensorboard

# tensorflow

# tensorflow-datasets

# tensorflow-hub

# tensorflow-metadata

# tfds-nightly

psutil==5.9.5

# via

# ipykernel

# portpicker

# tensorflow-datasets

# tfds-nightly

ptyprocess==0.7.0

# via

# pexpect

# terminado

pure-eval==0.2.2

# via stack-data

pyasn1==0.5.0

# via

# pyasn1-modules

# rsa

pyasn1-modules==0.3.0

# via google-auth

pycparser==2.21

# via cffi

pyglove==0.4.2

# via

# paxml

# seqio

# seqio-nightly

pygments==2.16.1

# via

# ipython

# jupyter-console

# nbconvert

# qtconsole

# rich

pyparsing==3.0.9

# via matplotlib

python-dateutil==2.8.2

# via

# arrow

# jupyter-client

# matplotlib

# pandas

python-json-logger==2.0.7

# via jupyter-events

pytz==2023.3

# via pandas

pyyaml==6.0.1

# via

# flax

# huggingface-hub

# jupyter-events

# libcst

# ml-collections

# orbax-checkpoint

# transformers

pyzmq==25.1.1

# via

# ipykernel

# jupyter-client

# jupyter-console

# jupyter-server

# qtconsole

qtconsole==5.4.3

# via jupyter

qtpy==2.3.1

# via qtconsole

referencing==0.30.2

# via

# jsonschema

# jsonschema-specifications

# jupyter-events

regex==2023.8.8

# via

# nltk

# sacrebleu

# transformers

requests==2.31.0

# via

# huggingface-hub

# jax

# jupyterlab-server

# requests-oauthlib

# tensorboard

# tensorflow-datasets

# tfds-nightly

# transformers

requests-oauthlib==1.3.1

# via google-auth-oauthlib

rfc3339-validator==0.1.4

# via

# jsonschema

# jupyter-events

rfc3986-validator==0.1.1

# via

# jsonschema

# jupyter-events

rich==13.5.2

# via flax

rouge-score==0.1.2

# via t5

rpds-py==0.9.2

# via

# jsonschema

# referencing

rsa==4.9

# via google-auth

sacrebleu==2.3.1

# via t5

safetensors==0.3.3

# via transformers

scikit-learn==1.3.0

# via

# lingvo

# t5

scipy==1.11.2

# via

# jax

# jaxlib

# mesh-tensorflow

# scikit-learn

# t5

send2trash==1.8.2

# via jupyter-server

sentencepiece==0.1.99

# via

# lingvo

# seqio

# seqio-nightly

# t5

seqio==0.0.16

# via -r requirements.in

seqio-nightly==0.0.15.dev20230825

# via

# paxml

# t5

six==1.16.0

# via

# asttokens

# astunparse

# bleach

# google-auth

# google-pasta

# jupyter-http-over-ws

# keras-preprocessing

# mesh-tensorflow

# ml-collections

# promise

# python-dateutil

# rfc3339-validator

# rouge-score

# t5

# tensorflow

sniffio==1.3.0

# via anyio

soupsieve==2.4.1

# via beautifulsoup4

stack-data==0.6.2

# via ipython

sympy==1.12

# via lingvo

t5==0.9.4

# via paxml

tabulate==0.9.0

# via sacrebleu

tensorboard==2.9.1

# via tensorflow

tensorboard-data-server==0.6.1

# via tensorboard

tensorboard-plugin-wit==1.8.1

# via tensorboard

tensorflow==2.9.3

# via

# lingvo

# paxml

# praxis

# tensorflow-text

tensorflow-datasets==4.8.3

# via

# lingvo

# mesh-tensorflow

# paxml

# praxis

tensorflow-estimator==2.9.0

# via tensorflow

tensorflow-hub==0.14.0

# via

# lingvo

# tensorflow-text

tensorflow-io-gcs-filesystem==0.33.0

# via tensorflow

tensorflow-metadata==1.12.0

# via

# paxml

# praxis

# tensorflow-datasets

# tfds-nightly

tensorflow-text==2.9.0

# via

# lingvo

# paxml

# praxis

# seqio

# seqio-nightly

tensorstore==0.1.41

# via

# flax

# orbax-checkpoint

# paxml

termcolor==2.3.0

# via

# tensorflow

# tensorflow-datasets

# tfds-nightly

terminado==0.17.1

# via

# jupyter-server

# jupyter-server-terminals

tfds-nightly==4.8.3.dev202303280045

# via

# paxml

# praxis

# seqio

# seqio-nightly

# t5

threadpoolctl==3.2.0

# via scikit-learn

tinycss2==1.2.1

# via nbconvert

tokenizers==0.13.3

# via transformers

toml==0.10.2

# via

# tensorflow-datasets

# tfds-nightly

tomli==2.0.1

# via jupyterlab

toolz==0.12.0

# via chex

tornado==6.3.3

# via

# ipykernel

# jupyter-client

# jupyter-http-over-ws

# jupyter-server

# jupyterlab

# notebook

# terminado

tqdm==4.66.1

# via

# huggingface-hub

# nltk

# tensorflow-datasets

# tfds-nightly

# transformers

traitlets==5.9.0

# via

# comm

# ipykernel

# ipython

# ipywidgets

# jupyter-client

# jupyter-console

# jupyter-core

# jupyter-events

# jupyter-server

# jupyterlab

# matplotlib-inline

# nbclient

# nbconvert

# nbformat

# qtconsole

transformers==4.32.0

# via t5

typeguard==4.1.2

# via

# jaxtyping

# praxis

typing-extensions==4.7.1

# via

# async-lru

# chex

# clu

# etils

# fiddle

# flax

# huggingface-hub

# jaxtyping

# libcst

# orbax-checkpoint

# tensorflow

# typeguard

# typing-inspect

typing-inspect==0.9.0

# via

# dataclasses-json

# libcst

tzdata==2023.3

# via pandas

uri-template==1.3.0

# via jsonschema

urllib3==1.26.16

# via

# google-auth

# requests

wcwidth==0.2.6

# via prompt-toolkit

webcolors==1.13

# via jsonschema

webencodings==0.5.1

# via

# bleach

# tinycss2

websocket-client==1.6.2

# via jupyter-server

werkzeug==2.3.7

# via tensorboard

wheel==0.41.2

# via

# astunparse

# tensorboard

widgetsnbextension==4.0.8

# via ipywidgets

wrapt==1.15.0

# via

# clu

# tensorflow

# tensorflow-datasets

# tfds-nightly

zipp==3.16.2

# via

# etils

# importlib-metadata

# The following packages are considered to be unsafe in a requirements file:

setuptools==68.1.2

# via

# tensorboard

# tensorflow