[Do Not Merge] Testing deps in maxdiffusion image built using JAII

requirements.txt

@@ -1,38 +1,185 @@
---extra-index-url https://download.pytorch.org/whl/cpu
-jax>=0.6.2
-jaxlib>=0.4.30
-grain
-google-cloud-storage>=2.17.0
-absl-py
-datasets
-flax>=0.10.2
-optax>=0.2.3
-torch>=2.6.0
-torchvision>=0.20.1
-ftfy
-tensorboard>=2.17.0
-tensorboardx>=2.6.2.2
-tensorboard-plugin-profile>=2.15.2
-Jinja2
-scikit-image
-parameterized
-Pillow
-pylint
-pyink
-pytest==8.2.2
-tensorflow>=2.17.0
-tensorflow-datasets>=4.9.6
-ruff>=0.1.5,<=0.2
-git+https://github.com/Lightricks/LTX-Video
-git+https://github.com/zmelumian972/xla@torchax/jittable_module_callable#subdirectory=torchax
-opencv-python-headless==4.10.0.84
-orbax-checkpoint
-tokenizers==0.21.0
-huggingface_hub>=0.30.2
-transformers==4.48.1
+absl-py==2.3.1
+aiohappyeyeballs==2.6.1
+aiohttp==3.12.14
+aiosignal==1.4.0
+annotated-types==0.7.0
+anyio==4.9.0
+aqtp==0.8.4
+array_record==0.7.2
+astroid==3.3.11
+astunparse==1.6.3
+attrs==25.3.0
+black==24.10.0
+cachetools==5.5.2
+certifi==2025.7.14
+charset-normalizer==3.4.2
+cheroot==10.0.1
+chex==0.1.90
+click==8.2.1
+cloud-accelerator-diagnostics==0.1.1
+cloud-tpu-diagnostics==0.1.5
+cloudpickle==3.1.1
+clu==0.0.12
+datasets==4.0.0
+decorator==5.2.1
+dill==0.3.8
+dm-tree==0.1.9
+docstring_parser==0.17.0
 einops==0.8.0
-sentencepiece
-aqtp
+etils==1.13.0
+fiddle==0.3.0
+filelock==3.18.0
+flatbuffers==25.2.10
+flax==0.10.6
+frozenlist==1.7.0
+fsspec==2025.3.0
+ftfy==6.3.1
+gast==0.6.0
+gcsfs==2025.3.0
+google-api-core==2.25.1
+google-auth==2.40.3
+google-auth-oauthlib==1.2.2
+google-cloud-aiplatform==1.105.0
+google-cloud-appengine-logging==1.6.2
+google-cloud-audit-log==0.3.2
+google-cloud-bigquery==3.35.1
+google-cloud-core==2.4.3
+google-cloud-logging==3.12.1
+google-cloud-monitoring==2.27.2
+google-cloud-resource-manager==1.14.2
+google-cloud-storage==2.19.0
+google-crc32c==1.7.1
+google-genai==1.27.0
+google-pasta==0.2.0
+google-resumable-media==2.7.2
+googleapis-common-protos==1.70.0
+grain==0.2.11
+graphviz==0.21
+grpc-google-iam-v1==0.14.2
+grpcio==1.74.0
+grpcio-status==1.74.0
+gviz-api==1.10.0
+h11==0.16.0
+h5py==3.14.0
+hf-xet==1.1.5
+hf_transfer==0.1.9
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.34.1
+humanize==4.12.3
+idna==3.10
 imageio==2.37.0
 imageio-ffmpeg==0.6.0
-hf_transfer>=0.1.9
+immutabledict==4.2.1
+importlib_metadata==8.7.0
+importlib_resources==6.5.2
+iniconfig==2.1.0
+isort==6.0.1
+jaraco.functools==4.2.1
+jax==0.7.0
+jaxlib==0.7.0
+Jinja2==3.1.6
+keras==3.10.0
+lazy_loader==0.4
+libclang==18.1.1
+libcst==1.8.2
+libtpu==0.0.19
+Markdown==3.8.2
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+mccabe==0.7.0
+mdurl==0.1.2
+ml_collections==1.1.0
+ml_dtypes==0.5.1
+ml_goodput_measurement==0.0.13
+more-itertools==10.7.0
+mpmath==1.3.0
+msgpack==1.1.1
+multidict==6.6.3
+multiprocess==0.70.16
+mypy_extensions==1.1.0
+namex==0.1.0
+nest-asyncio==1.6.0
+networkx==3.5
+numpy==2.1.3
+oauthlib==3.3.1
+opencv-python-headless==4.10.0.84
+opentelemetry-api==1.35.0
+opt_einsum==3.4.0
+optax==0.2.5
+optree==0.17.0
+orbax-checkpoint==0.11.19
+packaging==25.0
+pandas==2.3.1
+parameterized==0.9.0
+pathspec==0.12.1
+pillow==11.3.0
+platformdirs==4.3.8
+pluggy==1.6.0
+promise==2.3
+propcache==0.3.2
+proto-plus==1.26.1
+protobuf==5.29.5
+psutil==7.0.0
+pyarrow==21.0.0
+pyasn1==0.6.1
+pyasn1_modules==0.4.2
+pydantic==2.11.7
+pydantic_core==2.33.2
+Pygments==2.19.2
+pyink==24.10.1
+pylint==3.3.7
+pytest==8.2.2
+python-dateutil==2.9.0.post0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.4
+requests-oauthlib==2.0.0
+rich==14.1.0
+rsa==4.9.1
+ruff==0.2.0
+safetensors==0.5.3
+scikit-image==0.25.2
+scipy==1.16.0
+sentencepiece==0.2.0
+setuptools==80.9.0
+shapely==2.1.1
+simple-parsing==0.1.7
+simplejson==3.20.1
+six==1.17.0
+sniffio==1.3.1
+sympy==1.14.0
+tenacity==8.5.0
+tensorboard==2.19.0
+tensorboard-data-server==0.7.2
+tensorboard_plugin_profile==2.20.4
+tensorboardX==2.6.4
+tensorflow==2.19.0
+tensorflow-datasets==4.9.9
+tensorflow-metadata==1.17.2
+tensorstore==0.1.76
+termcolor==3.1.0
+tifffile==2025.6.11
+tokenizers==0.21.0
+toml==0.10.2
+tomlkit==0.13.3
+toolz==1.0.0
+torch==2.7.1+cpu
+torchvision==0.22.1+cpu
+tqdm==4.67.1
+transformers==4.48.1
+treescope==0.1.9
+typing-inspection==0.4.1
+typing_extensions==4.14.1
+tzdata==2025.2
+urllib3==2.5.0
+wcwidth==0.2.13
+websockets==15.0.1
+Werkzeug==3.1.3
+wheel==0.45.1
+wrapt==1.17.2
+xprof==2.20.4
+xxhash==3.5.0
+yarl==1.20.1
+zipp==3.23.0