Skip to content

Optimize MaxText cross-entropy loss memory, vocabulary tiling, and PyTree engine caching: - #5201

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_978486370
Open

Optimize MaxText cross-entropy loss memory, vocabulary tiling, and PyTree engine caching:#5201
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_978486370

Commits

Commits on Sep 11, 2026