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 407dc774f7 [Mosaic TPU] Support all cases for extui. 
PiperOrigin-RevId: 678331795
2024-09-24 11:35:03 -07:00
..
dialect/tpu
[Mosaic TPU] Support all cases for extui.
2024-09-24 11:35:03 -07:00
gpu
Add basic PyTorch integration for Mosaic GPU
2024-09-18 12:55:23 +00:00
python
Clean up BUILD files.
2024-08-26 09:11:17 -07:00
BUILD
Clean up BUILD files.
2024-08-26 09:11:17 -07:00