2 docs tagged with "jit"

JAX jit, grad, vmap, pmap - functional transformations for high-performance ML, XLA compilation, and NumPy-compatible ML research.

LLVM-based JIT compilation for Python numerical code - GPU acceleration, parallel loops, and ufunc creation with @jit and @cuda.jit.