mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,442 Commits 500 Branches 263 Tags
Commit Graph

1 Commits

Author SHA1 Message Date
Peter Hawkins
78543f7bb8 Add jax.extend.mlir. 
Some users of JAX want to use the MLIR dialects defined in jaxlib. In particular, these need to be used by custom lowering rules. Add a semi-public (jax.extend) API to access these, rather than having them use jax._src.lib.mlir.

PiperOrigin-RevId: 588448489
 2023-12-06 09:16:43 -08:00