jax
2d69bd68 - Add MPMD to IFRT IR lowerings to jaxlib._sdy_mpmd

Commit

109 days ago

Add MPMD to IFRT IR lowerings to jaxlib._sdy_mpmd PiperOrigin-RevId: 818713630

References

#32466 - Add MPMD to IFRT IR lowerings to jaxlib._sdy_mpmd

Author

JoelWee

JoelWee

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading