jax
a7fe545c - [pltpu.emit_pipeline] Support async pipeline prefetch.

Commit

61 days ago

[pltpu.emit_pipeline] Support async pipeline prefetch. PiperOrigin-RevId: 981514857

References

Author

rdyro

Committer

Google-ML-Automation

Parents