PhilipVinc / numba4jax
Use numba-compiled kernels from within Jax
☆25Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for numba4jax
- ☆49Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Stencil computations in JAX☆66Updated last year
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- A dead simple Python package for creating custom JAX pytree objects☆62Updated last week
- JAX + Units☆43Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- ☆29Updated 2 years ago
- Abstract your array operations.☆137Updated last month
- Multiple dispatch over abstract array types in JAX.