Skip to content

Commit

Permalink
[Pallas] Refactor Mosaic lowering to encapsulate jaxpr->mlir type cre…
Browse files Browse the repository at this point in the history 
…ation all in one place

PiperOrigin-RevId: 583532870
  • Loading branch information
@sharadmv
sharadmv authored and jax authors committed Nov 18, 2023
1 parent 3872955 commit df9dd53
Showing 1 changed file with 249 additions and 217 deletions.

0 comments on commit df9dd53

Please sign in to comment.