mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
Jevin Jiang 6e37c4202d [Mosaic] Add support for sublane strided store. 
PiperOrigin-RevId: 552595663
2023-07-31 14:40:27 -07:00
..
dialect/tpu
[Mosaic] Add support for sublane strided store.
2023-07-31 14:40:27 -07:00
python
[Mosaic] Add support for sublane strided store.
2023-07-31 14:40:27 -07:00
BUILD
Enable building jaxlib w/ Mosaic
2023-07-26 03:59:30 -07:00