jax
9c76aa51 - [pallas:triton] Migrated example kernels to `plgpu.{load,store}`

Commit

216 days ago

[pallas:triton] Migrated example kernels to `plgpu.{load,store}` PiperOrigin-RevId: 786856202

References

Author

superbobry

Committer

Google-ML-Automation

Parents