jax
d07d642d - Merge pull request #27177 from jax-ml:mixing_modes

Commit

1 year ago

Merge pull request #27177 from jax-ml:mixing_modes PiperOrigin-RevId: 737047069

References

#27177 - Add an example to the sharding-in-types doc about mixing modes

#292 - CI: 03/17/25 upstream sync

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading