Merge pull request #2898 from AI-Hypercomputer:tests_docker_image

PiperOrigin-RevId: 850456883

Introduce a DSv3 config with 2 logical fsdp sharding axes.

PiperOrigin-RevId: 843675516

Merge pull request #2706 from AI-Hypercomputer:mohit/tokamax_quant_gmm

PiperOrigin-RevId: 834605168

Merge pull request #2676 from AI-Hypercomputer:pypi_release

PiperOrigin-RevId: 832378885

Merge pull request #2539 from AI-Hypercomputer:qinwen/latest-tokamax

PiperOrigin-RevId: 823749360

Merge pull request #2538 from AI-Hypercomputer:mohit/fix_docker

PiperOrigin-RevId: 822796389

Merge pull request #2488 from AI-Hypercomputer:carlosbus/training_v6e…

…_gemma3_12b

PiperOrigin-RevId: 820979201

maxtext v0.1.0 release

Use this release for [tpu-recipes](https://github.com/AI-Hypercompute…

…r/tpu-recipes) that require version `tpu-recipes-v0.1.4`

Merge pull request #1821 from AI-Hypercomputer:yixuannwang-test2

PiperOrigin-RevId: 771593926