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-15 19:36:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jaxlib
/
mosaic
/
python
History
Adam Paszke
64433435ff
Fix OSS build for the Mosaic GPU dialect
2025-01-02 15:55:03 +00:00
..
BUILD
[MOSAIC:GPU] Add
async_load
,
async_store
, and supporting attributes to the MLIR Mosaic GPU Dialect.
2024-11-08 14:34:23 -08:00
layout_defs.py
[Mosaic] Remove Python implementation of apply_vector_layout and infer_memref_layout.
2024-01-10 13:00:21 -08:00
mosaic_gpu.py
Fix OSS build for the Mosaic GPU dialect
2025-01-02 15:55:03 +00:00
tpu_python.td
Integrate LLVM at
llvm/llvm-project@571c1292b6
2023-07-21 15:56:28 -07:00
tpu.py
[Pallas/Mosaic] Add support for returning values from run_scoped
2024-07-11 18:37:09 -07:00