jax
108f08d2 - [Pallas] Enable mpmd with pl.kernel

Commit

9 days ago

[Pallas] Enable mpmd with pl.kernel PiperOrigin-RevId: 907073016

References

#37177 - [Pallas] Enable mpmd with pl.kernel

Author

sharadmv

sharadmv

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading