jax
7c5871f4 - [Pallas TPU] Hoist prologue and epilogue outside of pipeline loop

Commit

321 days ago

[Pallas TPU] Hoist prologue and epilogue outside of pipeline loop PiperOrigin-RevId: 738038138

References

Author

Google-ML-Automation

Committer

Google-ML-Automation

Parents