|
|
|
|
|
|
by 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). |
|
|