training_engine: add stage_batch with host-buffered multi-batch queue (TUNIX_STAGE_MAX_DEVICE_BATCHES) by A9isha · Pull Request #5751 · AI-Hypercomputer/maxtext

✅ 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.

@A9isha

The following contributors were found for this pull request:

✅ 066a2a7 Author: @A9isha <ma*****ra@google.com>

(Only the first commit for a unique contributor is listed.)