Hacker News Reader

Dex is also missing user authored composable program transformations, which is one of JAX’s hidden superpowers.

So not quite “JAX without limitations” — but certainly without some of the limitations.

This is both its strength and its weakness. As soon as you write a jaxpr interpreter, you lose all the tooling that makes the python interpreter so mature. For example stack traces and debugging become black holes. If jax made it easy to write these transformations without losing python’s benefits it would be incredible.

Are you talking about custom VJPs/JVPs?

No, I'm talking about custom `Jaxpr` interpreters which can modify programs to do things.

Reply on news.ycombinator.com