Implement host-pinned transfer microbenchmarks and PyTree checkpoint integrations for Orbax benchmarks.#3380
Open
copybara-service[bot] wants to merge 1 commit into
Open
Implement host-pinned transfer microbenchmarks and PyTree checkpoint integrations for Orbax benchmarks.#3380copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into
