# This file was autogenerated by uv via the following command:
#    uv pip compile pyproject.toml --extra tests -o requirements.txt
absl-py==2.1.0
    # via
    #   chex
    #   optax
    #   orbax-checkpoint
    #   tensorboard
    #   tensorflow
aiohttp==3.9.5
    # via gcsfs
aiosignal==1.3.1
    # via aiohttp
annotated-types==0.6.0
    # via pydantic
astunparse==1.6.3
    # via tensorflow
async-timeout==4.0.3
    # via aiohttp
attrs==23.2.0
    # via aiohttp
cachetools==5.3.3
    # via google-auth
certifi==2024.2.2
    # via requests
charset-normalizer==3.3.2
    # via requests
chex==0.1.86
    # via optax
decorator==5.1.1
    # via gcsfs
diffprivlib==0.6.4
    # via sarus-statistics
etils==1.5.2
    # via orbax-checkpoint
exceptiongroup==1.2.0
    # via pytest
filelock==3.13.4
    # via
    #   huggingface-hub
    #   transformers
flatbuffers==24.3.25
    # via tensorflow
flax==0.8.2
    # via sarus-synthetic-data (pyproject.toml)
frozenlist==1.4.1
    # via
    #   aiohttp
    #   aiosignal
fsspec==2024.3.1
    # via
    #   etils
    #   gcsfs
    #   huggingface-hub
gast==0.5.4
    # via tensorflow
gcsfs==2024.3.1
    # via sarus-synthetic-data (pyproject.toml)
google-api-core==2.18.0
    # via
    #   google-cloud-core
    #   google-cloud-storage
google-auth==2.29.0
    # via
    #   gcsfs
    #   google-api-core
    #   google-auth-oauthlib
    #   google-cloud-core
    #   google-cloud-storage
    #   tensorboard
google-auth-oauthlib==1.2.0
    # via
    #   gcsfs
    #   tensorboard
google-cloud-core==2.4.1
    # via google-cloud-storage
google-cloud-storage==2.16.0
    # via gcsfs
google-crc32c==1.5.0
    # via
    #   google-cloud-storage
    #   google-resumable-media
google-pasta==0.2.0
    # via tensorflow
google-resumable-media==2.7.0
    # via google-cloud-storage
googleapis-common-protos==1.63.0
    # via google-api-core
greenlet==3.0.3
    # via sqlalchemy
grpcio==1.63.0rc1
    # via
    #   tensorboard
    #   tensorflow
h5py==3.11.0
    # via tensorflow
huggingface-hub==0.24.5
    # via
    #   tokenizers
    #   transformers
idna==3.7
    # via
    #   requests
    #   yarl
importlib-metadata==7.1.0
    # via
    #   jax
    #   markdown
importlib-resources==6.4.0
    # via etils
iniconfig==2.0.0
    # via pytest
jax==0.4.25
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   chex
    #   flax
    #   optax
    #   orbax-checkpoint
jaxlib==0.4.25
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   chex
    #   optax
    #   orbax-checkpoint
joblib==1.4.0
    # via
    #   diffprivlib
    #   scikit-learn
keras==2.15.0
    # via tensorflow
libclang==18.1.1
    # via tensorflow
markdown==3.6
    # 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-dtypes==0.3.2
    # via
    #   jax
    #   jaxlib
    #   tensorflow
    #   tensorstore
msgpack==1.0.8
    # via
    #   flax
    #   orbax-checkpoint
multidict==6.0.5
    # via
    #   aiohttp
    #   yarl
nest-asyncio==1.6.0
    # via orbax-checkpoint
numpy==1.26.4
    # via
    #   chex
    #   diffprivlib
    #   flax
    #   h5py
    #   jax
    #   jaxlib
    #   ml-dtypes
    #   opt-einsum
    #   optax
    #   orbax-checkpoint
    #   pandas
    #   patsy
    #   pyarrow
    #   sarus-statistics
    #   scikit-learn
    #   scipy
    #   statsmodels
    #   tensorboard
    #   tensorflow
    #   tensorstore
    #   transformers
oauthlib==3.2.2
    # via requests-oauthlib
opendp==0.6.2
    # via sarus-statistics
opt-einsum==3.3.0
    # via
    #   jax
    #   tensorflow
optax==0.1.9
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   flax
orbax-checkpoint==0.4.3
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   flax
packaging==24.0
    # via
    #   huggingface-hub
    #   pytest
    #   statsmodels
    #   tensorflow
    #   transformers
pandas==1.4.4
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   sarus-statistics
    #   statsmodels
patsy==0.5.6
    # via statsmodels
pluggy==1.4.0
    # via pytest
proto-plus==1.23.0
    # via google-api-core
protobuf==3.20.3
    # via
    #   google-api-core
    #   googleapis-common-protos
    #   orbax-checkpoint
    #   proto-plus
    #   tensorboard
    #   tensorflow
psutil==5.9.8
    # via sarus-synthetic-data (pyproject.toml)
pyarrow==15.0.2
    # via sarus-synthetic-data (pyproject.toml)
pyasn1==0.6.0
    # via
    #   pyasn1-modules
    #   rsa
pyasn1-modules==0.4.0
    # via google-auth
pydantic==2.7.1
    # via sarus-synthetic-data (pyproject.toml)
pydantic-core==2.18.2
    # via pydantic
pygments==2.17.2
    # via rich
pytest==8.1.1
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   pytest-mock
pytest-mock==3.14.0
    # via sarus-synthetic-data (pyproject.toml)
python-dateutil==2.9.0.post0
    # via pandas
pytz==2024.1
    # via pandas
pyyaml==6.0.1
    # via
    #   flax
    #   huggingface-hub
    #   orbax-checkpoint
    #   transformers
regex==2023.12.25
    # via transformers
requests==2.31.0
    # via
    #   gcsfs
    #   google-api-core
    #   google-cloud-storage
    #   huggingface-hub
    #   requests-oauthlib
    #   tensorboard
    #   transformers
requests-oauthlib==2.0.0
    # via google-auth-oauthlib
rich==13.7.1
    # via flax
rsa==4.9
    # via google-auth
safetensors==0.4.3
    # via transformers
sarus-statistics==4.0.0
    # via sarus-synthetic-data (pyproject.toml)
scikit-learn==1.4.2
    # via
    #   diffprivlib
    #   sarus-statistics
scipy==1.13.0
    # via
    #   sarus-synthetic-data (pyproject.toml)
    #   diffprivlib
    #   jax
    #   jaxlib
    #   sarus-statistics
    #   scikit-learn
    #   statsmodels
setuptools==69.5.1
    # via
    #   diffprivlib
    #   tensorboard
    #   tensorflow
six==1.16.0
    # via
    #   astunparse
    #   google-pasta
    #   patsy
    #   python-dateutil
    #   tensorboard
    #   tensorflow
sqlalchemy==2.0.29
    # via sarus-statistics
statsmodels==0.14.1
    # via sarus-statistics
tensorboard==2.15.2
    # via tensorflow
tensorboard-data-server==0.7.2
    # via tensorboard
tensorflow==2.15.1
    # via sarus-synthetic-data (pyproject.toml)
tensorflow-estimator==2.15.0
    # via tensorflow
tensorflow-io-gcs-filesystem==0.36.0
    # via tensorflow
tensorstore==0.1.56
    # via
    #   flax
    #   orbax-checkpoint
termcolor==2.4.0
    # via tensorflow
threadpoolctl==3.4.0
    # via scikit-learn
tokenizers==0.19.1
    # via transformers
tomli==2.0.1
    # via pytest
toolz==0.12.1
    # via chex
tqdm==4.58.0
    # via
    #   huggingface-hub
    #   transformers
transformers==4.43.4
    # via sarus-synthetic-data (pyproject.toml)
types-psutil==5.9.5.20240423
    # via sarus-synthetic-data (pyproject.toml)
typing-extensions==4.11.0
    # via
    #   chex
    #   etils
    #   flax
    #   huggingface-hub
    #   orbax-checkpoint
    #   pydantic
    #   pydantic-core
    #   sqlalchemy
    #   tensorflow
urllib3==2.2.1
    # via requests
werkzeug==3.0.2
    # via tensorboard
wheel==0.43.0
    # via astunparse
wrapt==1.14.1
    # via tensorflow
yarl==1.9.4
    # via aiohttp
zipp==3.18.1
    # via
    #   etils
    #   importlib-metadata
    #   importlib-resources