jax
701c63e1 - [Pallas/TPU] Add API for megacore partitioning of pipelines

Commit

1 year ago

[Pallas/TPU] Add API for megacore partitioning of pipelines PiperOrigin-RevId: 644184524

References

Author

sharadmv

Committer

a-googler

Parents