absl-py==1.4.0
addict==2.4.0
appdirs==1.4.4
aqtp==0.1.0
array-record==0.4.1
astunparse==1.6.3
black==23.7.0
cachetools==5.3.1
certifi==2023.7.22
charset-normalizer==3.2.0
chex==0.1.7
click==8.1.6
cloudpickle==2.2.1
clu==0.0.9
contextlib2==21.6.0
contourpy==1.1.0
cycler==0.11.0
dacite==1.8.1
decorator==5.1.1
dm-haiku==0.0.9
dm-tree==0.1.8
docker-pycreds==0.4.0
docstring-parser==0.15
editdistance==0.6.2
einops==0.6.1
etils==1.4.1
filelock==3.13.4
flatbuffers==23.5.26
flax==0.7.2
flaxformer @ git+https://github.com/google/flaxformer@ba6089ba5d0602041bc1869375891adc1117d705
fonttools==4.42.0
fsspec==2024.3.1
gast==0.4.0
gin-config==0.5.0
gitdb==4.0.10
GitPython==3.1.32
google-auth==2.22.0
google-auth-oauthlib==1.0.0
google-pasta==0.2.0
googleapis-common-protos==1.60.0
grpcio==1.57.0
h5py==3.9.0
idna==3.4
importlib-metadata==6.8.0
importlib-resources==6.0.1
jax==0.4.13
jaxlib==0.4.13+cuda11.cudnn86
Jinja2==3.1.3
jmp==0.0.4
keras==2.12.0
kiwisolver==1.4.4
learned_optimization
libclang==16.0.6
Markdown==3.4.4
markdown-it-py==3.0.0
MarkupSafe==2.1.3
matplotlib==3.7.2
mdurl==0.1.2
ml-dtypes==0.3.1
ml_collections==0.1.1
mmengine==0.8.4
mpmath==1.3.0
msgpack==1.0.5
mypy-extensions==1.0.0
nest-asyncio==1.5.7
networkx==3.2.1
nose==1.3.7
numpy==1.24.3
nvidia-cublas-cu11==11.11.3.6
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu11==11.8.87
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvcc-cu11==11.8.89
nvidia-cuda-nvrtc-cu11==11.8.89
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu11==11.8.89
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu11==8.9.4.25
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu11==10.9.0.58
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu11==11.4.1.48
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu11==11.7.5.86
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.19.3
nvidia-nvjitlink-cu12==12.4.127
nvidia-nvtx-cu12==12.1.105
nvidia-pyindex==1.0.9
oauthlib==3.2.2
opencv-python==4.8.0.76
opt-einsum==3.3.0
optax==0.1.7
orbax-checkpoint==0.3.2
oryx==0.0.0.post1
packaging==23.1
pandas==2.1.1
pathspec==0.11.2
pathtools==0.1.2
Pillow==10.0.0
platformdirs==3.10.0
promise==2.3
protobuf==3.20.3
psutil==5.9.5
pyasn1==0.5.0
pyasn1-modules==0.3.0
pyglove==0.4.3
Pygments==2.16.1
pyparsing==3.0.9
python-dateutil==2.8.2
pytz==2023.3.post1
PyYAML==6.0.1
requests==2.31.0
requests-oauthlib==1.3.1
rich==13.5.2
rsa==4.9
scipy==1.9.3
sentencepiece==0.1.99
sentry-sdk==1.29.2
seqio==0.0.18
setproctitle==1.3.2
six==1.16.0
smart-open==7.0.4
smmap==5.0.0
sympy==1.12
tabulate==0.9.0
tensorboard==2.12.3
tensorboard-data-server==0.7.1
tensorflow==2.12.1
tensorflow-datasets==4.9.2
tensorflow-estimator==2.12.0
tensorflow-hub==0.14.0
tensorflow-io-gcs-filesystem==0.33.0
tensorflow-metadata==1.14.0
tensorflow-probability==0.21.0
tensorflow-text==2.12.1
tensorstore==0.1.41
termcolor==2.3.0
tfds-nightly==4.9.3.dev202309220044
toml==0.10.2
tomli==2.0.1
toolz==0.12.0
torch==2.2.2
torchvision==0.17.2
tqdm==4.66.1
triton==2.2.0
typing_extensions==4.5.0
tzdata==2023.3
urllib3==1.26.16
-e git+https://github.com/google-research/vision_transformer@ac6e056f9da686895f9f0f6ac026d3b5a464e59e#egg=vit_jax
wandb==0.15.7
Werkzeug==2.3.6
wrapt==1.14.1
yapf==0.40.1
zipp==3.16.2
