Hacker News new | ask | show | jobs
by woadwarrior01 1585 days ago
Jax also has experimental support for persisting its JIT cache on the filesystem in the ‘jax.experimental.compilation_cache’ module.
1 comments
VHRanger 1585 days ago
How does jax compare to numba?
link
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