This is work in progress with currently no ambition to be usable!
My attempt to combine the transformations of Jax with the compiler of Numba. See jax-ml/jax#2126 . It transforms a Jaxpr into an ast node.
- Numba doesn't support dynamic tuple creation -- but requires tuples for many Numpy functions
- Numba doesn't support any of Numpys broadcasting functions