Standalone JetStream removes pinned deps (#129)

Author: JoeZijunZhou

requirements.in

@@ -1,351 +1,19 @@
-#
-# 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
-anyio==3.7.1
-    # via
-    #   fastapi
-    #   starlette
-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.7.4
-    # via requests
-charset-normalizer==3.3.2
-    # via requests
-chex==0.1.7
-    # via optax
-click==8.1.7
-    # via
-    #   tfds-nightly
-    #   uvicorn
-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
-    #   anyio
-    #   pytest
-fastapi==0.103.2
-    # via -r requirements.in
-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
-gviz-api==1.10.0
-    # via tensorboard-plugin-profile
-h11==0.14.0
-    # via uvicorn
-h5py==3.10.0
-    # via tensorflow
-idna==3.7
-    # via
-    #   anyio
-    #   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
-    #   tensorboard-plugin-profile
-    #   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
-pydantic==1.10.17
-    # via fastapi
-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
-    #   gviz-api
-    #   ml-collections
-    #   promise
-    #   tensorboard-plugin-profile
-    #   tensorflow
-sniffio==1.3.1
-    # via anyio
-starlette==0.27.0
-    # via fastapi
-tensorboard==2.13.0
-    # via tensorflow
-tensorboard-data-server==0.7.2
-    # via tensorboard
-tensorboard-plugin-profile==2.15.1
-    # via -r requirements.in
-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
-    #   fastapi
-    #   flax
-    #   orbax-checkpoint
-    #   pydantic
-    #   tensorflow
-    #   uvicorn
-urllib3==2.2.2
-    # via
-    #   blobfile
-    #   requests
-uvicorn==0.30.1
-    # via -r requirements.in
-werkzeug==3.0.1
-    # via
-    #   tensorboard
-    #   tensorboard-plugin-profile
-wheel==0.42.0
-    # via
-    #   astunparse
-    #   tensorboard
-wrapt==1.16.0
-    # via
-    #   clu
-    #   tensorflow
-    #   tfds-nightly
-zipp==3.19.1
-    # via etils
-
-# The following packages are considered to be unsafe in a requirements file:
-# setuptools
+absl-py
+coverage
+flax
+grpcio
+jax
+jaxlib
+numpy
+portpicker
+prometheus-client
+pytest
+seqio
+tiktoken
+blobfile
+parameterized
+shortuuid
+fastapi
+uvicorn
+# For profiling
+tensorboard-plugin-profile