Commit cc5bbb5

authored and

Google-ML-Automation

committed

[pallas:mosaic] Enabled pytype for Mosaic GPU lowering

PiperOrigin-RevId: 781473222

1 parent a69deb8 commit cc5bbb5Copy full SHA for cc5bbb5

3 files changed

+124

-82

lines changed

jax/_src/pallas/mosaic_gpu

3 files changed

+124

-82

lines changed

Comments

(0)