requirements.txt

#
# This file is autogenerated by pip-compile with Python 3.10
# by the following command:
#
#    pip-compile requirements.in
#
absl-py==1.4.0
    # via
    #   -r requirements.in
    #   array-record
    #   chex
    #   clu
    #   etils
    #   ml-collections
    #   optax
    #   orbax-checkpoint
    #   seqio
    #   tensorboard
    #   tensorflow
    #   tensorflow-metadata
    #   tfds-nightly
array-record==0.5.0
    # via tfds-nightly
astunparse==1.6.3
    # via tensorflow
blobfile==2.1.1
    # via -r requirements.in
cachetools==5.3.2
    # via google-auth
certifi==2024.2.2
    # via requests
charset-normalizer==3.3.2
    # via requests
chex==0.1.7
    # via optax
click==8.1.7
    # via tfds-nightly
clu==0.0.10
    # via seqio
contextlib2==21.6.0
    # via ml-collections
coverage==7.4.4
    # via -r requirements.in
dm-tree==0.1.8
    # via
    #   chex
    #   tfds-nightly
docstring-parser==0.15
    # via pyglove
editdistance==0.6.2
    # via seqio
etils[array-types,enp,epath,epy,etqdm,etree]==1.6.0
    # via
    #   array-record
    #   clu
    #   orbax-checkpoint
    #   tfds-nightly
exceptiongroup==1.2.0
    # via pytest
filelock==3.14.0
    # via blobfile
flatbuffers==23.5.26
    # via tensorflow
flax==0.8.0
    # via
    #   -r requirements.in
    #   clu
fsspec==2023.12.2
    # via etils
gast==0.4.0
    # via tensorflow
google-auth==2.27.0
    # via
    #   google-auth-oauthlib
    #   tensorboard
google-auth-oauthlib==1.0.0
    # via tensorboard
google-pasta==0.2.0
    # via tensorflow
googleapis-common-protos==1.62.0
    # via tensorflow-metadata
grpcio==1.60.1
    # via
    #   -r requirements.in
    #   tensorboard
    #   tensorflow
h5py==3.10.0
    # via tensorflow
idna==3.7
    # via requests
importlib-resources==6.1.1
    # via etils
iniconfig==2.0.0
    # via pytest
jax==0.4.23
    # via
    #   -r requirements.in
    #   chex
    #   clu
    #   flax
    #   optax
    #   orbax-checkpoint
    #   seqio
jaxlib==0.4.23
    # via
    #   -r requirements.in
    #   chex
    #   clu
    #   optax
    #   orbax-checkpoint
    #   seqio
keras==2.13.1
    # via tensorflow
libclang==16.0.6
    # via tensorflow
lxml==4.9.4
    # via blobfile
markdown==3.5.2
    # via tensorboard
markdown-it-py==3.0.0
    # via rich
markupsafe==2.1.5
    # via werkzeug
mdurl==0.1.2
    # via markdown-it-py
ml-collections==0.1.1
    # via clu
ml-dtypes==0.3.2
    # via
    #   jax
    #   jaxlib
    #   tensorstore
msgpack==1.0.7
    # via
    #   flax
    #   orbax-checkpoint
nest-asyncio==1.6.0
    # via orbax-checkpoint
numpy==1.23.1
    # via
    #   -r requirements.in
    #   chex
    #   clu
    #   etils
    #   flax
    #   h5py
    #   jax
    #   jaxlib
    #   ml-dtypes
    #   opt-einsum
    #   optax
    #   orbax-checkpoint
    #   scipy
    #   seqio
    #   tensorboard
    #   tensorflow
    #   tensorflow-hub
    #   tensorstore
    #   tfds-nightly
oauthlib==3.2.2
    # via requests-oauthlib
opt-einsum==3.3.0
    # via
    #   jax
    #   tensorflow
optax==0.1.8
    # via flax
orbax-checkpoint==0.5.2
    # via flax
packaging==23.2
    # via
    #   clu
    #   pytest
    #   seqio
    #   tensorflow
parameterized==0.9.0
    # via -r requirements.in
pluggy==1.4.0
    # via pytest
portpicker==1.6.0
    # via -r requirements.in
prometheus-client==0.20.0
    # via -r requirements.in
promise==2.3
    # via tfds-nightly
protobuf==3.20.3
    # via
    #   googleapis-common-protos
    #   orbax-checkpoint
    #   seqio
    #   tensorboard
    #   tensorflow
    #   tensorflow-hub
    #   tensorflow-metadata
    #   tfds-nightly
psutil==5.9.8
    # via
    #   portpicker
    #   tfds-nightly
pyasn1==0.5.1
    # via
    #   pyasn1-modules
    #   rsa
pyasn1-modules==0.3.0
    # via google-auth
pycryptodomex==3.20.0
    # via blobfile
pyglove==0.4.4
    # via seqio
pygments==2.17.2
    # via rich
pytest==8.1.1
    # via -r requirements.in
pyyaml==6.0.1
    # via
    #   flax
    #   ml-collections
    #   orbax-checkpoint
regex==2024.4.28
    # via tiktoken
requests==2.32.0
    # via
    #   requests-oauthlib
    #   tensorboard
    #   tfds-nightly
    #   tiktoken
requests-oauthlib==1.3.1
    # via google-auth-oauthlib
rich==13.7.0
    # via flax
rsa==4.9
    # via google-auth
scipy==1.12.0
    # via
    #   jax
    #   jaxlib
sentencepiece==0.1.99
    # via seqio
seqio==0.0.19
    # via -r requirements.in
shortuuid==1.0.13
    # via -r requirements.in
six==1.16.0
    # via
    #   astunparse
    #   google-pasta
    #   ml-collections
    #   promise
    #   tensorflow
tensorboard==2.13.0
    # via tensorflow
tensorboard-data-server==0.7.2
    # via tensorboard
tensorflow==2.13.1
    # via tensorflow-text
tensorflow-estimator==2.13.0
    # via tensorflow
tensorflow-hub==0.16.1
    # via tensorflow-text
tensorflow-io-gcs-filesystem==0.35.0
    # via tensorflow
tensorflow-metadata==1.14.0
    # via tfds-nightly
tensorflow-text==2.13.0
    # via seqio
tensorstore==0.1.52
    # via
    #   flax
    #   orbax-checkpoint
termcolor==2.4.0
    # via
    #   tensorflow
    #   tfds-nightly
tf-keras==2.15.0
    # via tensorflow-hub
tfds-nightly==4.9.2.dev202308090034
    # via seqio
tiktoken==0.6.0
    # via -r requirements.in
toml==0.10.2
    # via tfds-nightly
tomli==2.0.1
    # via pytest
toolz==0.12.1
    # via chex
tqdm==4.66.3
    # via
    #   etils
    #   tfds-nightly
typing-extensions==4.5.0
    # via
    #   chex
    #   clu
    #   etils
    #   flax
    #   orbax-checkpoint
    #   tensorflow
urllib3==2.2.0
    # via
    #   blobfile
    #   requests
werkzeug==3.0.1
    # via tensorboard
wheel==0.42.0
    # via
    #   astunparse
    #   tensorboard
wrapt==1.16.0
    # via
    #   clu
    #   tensorflow
    #   tfds-nightly
zipp==3.17.0
    # via etils

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