numba4jax (pypi): Usa numba in jax-compiled kernels.
Keywords: Jax|Numba|compile|jit|kernel