google-research / dex-lang

Research language for array processing in the Haskell/ML family
BSD 3-Clause "New" or "Revised" License
1.56k stars 106 forks source link

Add a prototype translation from Jax to Dex in Haskell #1257

Closed axch closed 1 year ago

axch commented 1 year ago

To wit, serialize jaxprs from Python as JSON; parse them in Haskell into a new ADT modeling JAX IR; and compile to Dex SimpIR from there.

What works:

What doesn't work: