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 9d62d867bc [Mosaic] Add support for sublane strided load. 
PiperOrigin-RevId: 552581319
2023-07-31 13:47:43 -07:00
..
dialect/tpu
[Mosaic] Add support for sublane strided load.
2023-07-31 13:47:43 -07:00
python
[Mosaic] Add support for sublane strided load.
2023-07-31 13:47:43 -07:00
BUILD
Enable building jaxlib w/ Mosaic
2023-07-26 03:59:30 -07:00