jax
6348ed9f - [mosaic] Added a `k` prefix to `TPU_MemorySpace` members

Commit

225 days ago

[mosaic] Added a `k` prefix to `TPU_MemorySpace` members PiperOrigin-RevId: 772870655

References

Author

superbobry

Committer

charleshofer

Parents