This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
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
df4e0cbb73
[Mosaic] Add support for relayout from (8, 128) to (1, 128) with 32-bit data.
...
PiperOrigin-RevId: 553606429
2023-08-03 15:24:15 -07:00
..
dialect
/tpu
[Mosaic] Add a reinterpret cast for memrefs
2023-08-03 01:50:19 -07:00
python
[Mosaic] Add support for relayout from (8, 128) to (1, 128) with 32-bit data.
2023-08-03 15:24:15 -07:00
BUILD
Enable building jaxlib w/ Mosaic
2023-07-26 03:59:30 -07:00
