Optimize MaxText cross-entropy loss memory, vocabulary tiling, and PyTree engine caching: - #5201
Open
copybara-service[bot] wants to merge 1 commit into
Open
Optimize MaxText cross-entropy loss memory, vocabulary tiling, and PyTree engine caching:#5201copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into
Google CLA / cla/google
succeeded
Sep 11, 2026 in 15s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
The following contributors were found for this pull request:
✅ 2182c25 PR Opener: @copybara-service[bot]
✅ 2182c25 Author: @akushnarov <an****ku@google.com>
(Only the first commit for a unique contributor is listed.)
Loading