Skip to content

Upstream num_real_microbatches to micro_vmap and micro_grad and add unit tests. - #1570

Merged
copybara-service[bot] merged 1 commit into
mainfrom
test_859675520
Jan 22, 2026
Merged

Upstream num_real_microbatches to micro_vmap and micro_grad and add unit tests.#1570
copybara-service[bot] merged 1 commit into
mainfrom
test_859675520