3 ms·
How does jax compare to numba?
by VHRanger 5y ago
How does jax compare to numba?
- woadwarrior01 5y agonumba is more general. Any change to the shapes of arrays triggers a JIT recompilation in jax, numba is a bit more forgiving. jax has autodiff that numba doesn't. Also, JAX supports TPUs, which numba doesn't support (yet).