requirements.lock

# generated by rye
# use `rye lock` or `rye sync` to update this lockfile
#
# last locked with the following flags:
#   pre: false
#   features: []
#   all-features: false
#   with-sources: false
#   generate-hashes: false
#   universal: false

-e file:.
absl-py==2.1.0
    # via chex
    # via etils
    # via keras
    # via ml-collections
    # via optax
    # via orbax-checkpoint
    # via tensorboard
    # via tensorflow
    # via tensorflow-datasets
    # via tensorflow-metadata
    # via tensorflow-probability
astunparse==1.6.3
    # via tensorflow
certifi==2024.7.4
    # via requests
charset-normalizer==3.3.2
    # via requests
chex==0.1.86
    # via optax
click==8.1.7
    # via tensorflow-datasets
cloudpickle==3.0.0
    # via tensorflow-probability
contextlib2==21.6.0
    # via ml-collections
decorator==5.1.1
    # via tensorflow-probability
dm-tree==0.1.8
    # via tensorflow-datasets
    # via tensorflow-probability
docstring-parser==0.16
    # via simple-parsing
etils==1.9.2
    # via optax
    # via orbax-checkpoint
    # via tensorflow-datasets
flatbuffers==24.3.25
    # via tensorflow
flax==0.8.5
    # via rflow
fsspec==2024.6.1
    # via etils
gast==0.6.0
    # via tensorflow
    # via tensorflow-probability
google-pasta==0.2.0
    # via tensorflow
googleapis-common-protos==1.63.2
    # via tensorflow-metadata
grpcio==1.66.0
    # via tensorboard
    # via tensorflow
h5py==3.11.0
    # via keras
    # via tensorflow
humanize==4.10.0
    # via orbax-checkpoint
idna==3.8
    # via requests
immutabledict==4.2.0
    # via tensorflow-datasets
importlib-resources==6.4.4
    # via etils
jax==0.4.31
    # via chex
    # via flax
    # via optax
    # via orbax-checkpoint
    # via rflow
    # via tensorflow-probability
jaxlib==0.4.31
    # via chex
    # via jax
    # via optax
    # via orbax-checkpoint
    # via tensorflow-probability
keras==3.5.0
    # via tensorflow
libclang==18.1.1
    # via tensorflow
markdown==3.7
    # 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 rflow
ml-dtypes==0.4.0
    # via jax
    # via jaxlib
    # via keras
    # via tensorflow
    # via tensorstore
msgpack==1.0.8
    # via flax
    # via orbax-checkpoint
namex==0.0.8
    # via keras
nest-asyncio==1.6.0
    # via orbax-checkpoint
numpy==1.26.4
    # via chex
    # via etils
    # via flax
    # via h5py
    # via jax
    # via jaxlib
    # via keras
    # via ml-dtypes
    # via opt-einsum
    # via optax
    # via orbax-checkpoint
    # via pyarrow
    # via scipy
    # via tensorboard
    # via tensorflow
    # via tensorflow-datasets
    # via tensorflow-probability
    # via tensorstore
opt-einsum==3.3.0
    # via jax
    # via tensorflow
optax==0.2.3
    # via flax
    # via rflow
optree==0.12.1
    # via keras
orbax-checkpoint==0.6.1
    # via flax
packaging==24.1
    # via keras
    # via tensorboard
    # via tensorflow
promise==2.3
    # via tensorflow-datasets
protobuf==4.25.4
    # via googleapis-common-protos
    # via orbax-checkpoint
    # via tensorboard
    # via tensorflow
    # via tensorflow-datasets
    # via tensorflow-metadata
psutil==6.0.0
    # via tensorflow-datasets
pyarrow==17.0.0
    # via tensorflow-datasets
pygments==2.18.0
    # via rich
pyyaml==6.0.2
    # via flax
    # via ml-collections
    # via orbax-checkpoint
requests==2.32.3
    # via tensorflow
    # via tensorflow-datasets
rich==13.7.1
    # via flax
    # via keras
scipy==1.14.1
    # via jax
    # via jaxlib
setuptools==73.0.1
    # via tensorboard
    # via tensorflow
simple-parsing==0.1.5
    # via tensorflow-datasets
six==1.16.0
    # via astunparse
    # via google-pasta
    # via ml-collections
    # via promise
    # via tensorboard
    # via tensorflow
    # via tensorflow-probability
tensorboard==2.17.1
    # via tensorflow
tensorboard-data-server==0.7.2
    # via tensorboard
tensorflow==2.17.0
    # via rflow
tensorflow-datasets==4.9.6
    # via rflow
tensorflow-io-gcs-filesystem==0.37.1
    # via tensorflow
tensorflow-metadata==1.15.0
    # via tensorflow-datasets
tensorflow-probability==0.24.0
    # via rflow
tensorstore==0.1.64
    # via flax
    # via orbax-checkpoint
termcolor==2.4.0
    # via tensorflow
    # via tensorflow-datasets
toml==0.10.2
    # via tensorflow-datasets
toolz==0.12.1
    # via chex
tqdm==4.66.5
    # via etils
    # via tensorflow-datasets
typing-extensions==4.12.2
    # via chex
    # via etils
    # via flax
    # via optree
    # via orbax-checkpoint
    # via simple-parsing
    # via tensorflow
urllib3==2.2.2
    # via requests
werkzeug==3.0.4
    # via tensorboard
wheel==0.44.0
    # via astunparse
wrapt==1.16.0
    # via tensorflow
    # via tensorflow-datasets
zipp==3.20.0
    # via etils