jax
af5b5c8a - [Mosaic TPU] Add pack/unpack elementwise operations.

Commit

204 days ago

[Mosaic TPU] Add pack/unpack elementwise operations. PiperOrigin-RevId: 828183956

References

Author

megrez-yliu

Committer

Google-ML-Automation

Parents