News for hackers. Stuff that matters.

A comment on Faster Python calculations with Numba. Parent comment

Re:Faster Python calculations with Numba by VHRanger on Saturday February 19, 2022 @11:27AM (#30398268)
How does jax compare to numba?
[ Parent | on HN ]
Re:Faster Python calculations with Numba by woadwarrior01 on Saturday February 19, 2022 @02:58PM (#30400000)
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).
[ Parent | on HN ]