jax
8c351917 - Enable `jax.device_put` to a sharding with no local devices.

Commit

331 days ago

Enable `jax.device_put` to a sharding with no local devices. PiperOrigin-RevId: 737797815

References

Author

emilyfertig

Committer

Google-ML-Automation

Parents