jax

8ef49a3b - Don't turn on C++ jit by default until we're sure it works everywhere.