jit

JAX Crash Course - Accelerating Machine Learning code!

AssemblyAI

A JAX function transformation that applies Just-In-Time compilation to speed up Python and NumPy code.