jax
dfdb0dc6 - [Pallas:MGPU] Add a collective matmul tutorial

Commit

121 days ago

[Pallas:MGPU] Add a collective matmul tutorial PiperOrigin-RevId: 817683205

References

#32488 - [Pallas:MGPU] Add a collective matmul tutorial

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading