jax
e23ee15d - Merge pull request #19445 from jakevdp:full-device

Commit

2 years ago

Merge pull request #19445 from jakevdp:full-device PiperOrigin-RevId: 600502184

References

#19445 - lax.full: add sharding argument

Author

a-googler

a-googler

Parents

Loading