Hacker News new | ask | show | jobs
by VHRanger 1585 days ago
How does jax compare to numba?
1 comments
woadwarrior01 1585 days ago
numba 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).
link