jax
0e0fc0ac - [JAX] Add a test using inputs with different device orders for a single colocated Python call

Commit

1 year ago

[JAX] Add a test using inputs with different device orders for a single colocated Python call PiperOrigin-RevId: 708461989

References

Author

hyeontaek

Committer

Google-ML-Automation

Parents