Because JAX is not designed around a mature compiler stack. The history of Jax is more so that it matured alongside the compiler...
Because JAX is not designed around a mature compiler stack. The history of Jax is more so that it matured alongside the compiler...