jax
bc3a3f0b - [pallas:mosaic] Allowed registering lowering per `pltpu.KernelType`

Commit

302 days ago

[pallas:mosaic] Allowed registering lowering per `pltpu.KernelType` PiperOrigin-RevId: 757925207

Author

superbobry

Committer

Google-ML-Automation

Parents