diff --git a/requirements_with_jax_stable_stack_0.6.0.txt b/requirements_with_jax_stable_stack_0.6.0.txt new file mode 100644 index 0000000000..d7d233733a --- /dev/null +++ b/requirements_with_jax_stable_stack_0.6.0.txt @@ -0,0 +1,56 @@ +absl_py==2.2.2 +aqtp==0.9.0 +#benchmark_db_writer==1.0.0.dev20250610 +#benchmark_db_writer.egg==info +cloud_accelerator_diagnostics==0.1.1 +cloud_tpu_diagnostics==0.1.5 +chex==0.1.90 +datasets==3.6.0 +etils==1.12.2 +evaluate==0.4.4 +flax==0.10.7 +grain==0.2.10 +grpcio==1.72.0rc1 +huggingface_hub==0.33.0 +#jax==0.6.0 +#jaxlib==0.6.0 # Manually adding to ensure consistency in future +jaxtyping==0.3.2 +#jetstream==0.1.0 +jsonlines==4.0.0 +#libtpu==0.0.15 # Manually adding to ensure consistency in future +matplotlib==3.10.3 +ml_collections==1.1.0 +ml_dtypes==0.5.1 +ml_goodput_measurement==0.0.11 +nltk==3.9.1 +numpy +omegaconf==2.3.0 +optax==0.2.5 +orbax-checkpoint +pandas==2.3.0 +pathwaysutils@git+https://github.com/AI-Hypercomputer/pathways-utils.git@d6fffd3f9bd5e06c323f5bade04b40fa741c728f +Pillow==11.2.1 +protobuf +psutil==7.0.0 +pytest==8.4.1 +PyYAML==6.0.2 +PyYAML==6.0.2 +Requests==2.32.4 +qwix@git+https://github.com/google/qwix.git +safetensors==0.5.3 +sentencepiece==0.2.0 +setuptools +tabulate==0.9.0 +tensorboard_plugin_profile==2.13.0 +tensorboardX==2.6.4 +tensorflow==2.19.0 +tensorflow_datasets==4.9.9 +tensorflow_text==2.19.0 +tensorstore==0.1.75 +tiktoken==0.9.0 +#torch==2.7.1 +tqdm==4.67.1 +#transformer_engine==2.4.0 +transformers==4.52.4 +trl==0.19.0 +urllib3==2.5.0