jax
df6415b4 - [Pallas:MGPU] Allow inline_mgpu inside a warp context

Commit

3 days ago

[Pallas:MGPU] Allow inline_mgpu inside a warp context PiperOrigin-RevId: 876128638

References

Author

apaszke

Committer

Google-ML-Automation

Parents