RESULTS
NumPy:          0.6294 seconds
JAX (no JIT):   0.0094 seconds
JAX (with JIT): 0.0198 seconds

Speedup (JAX vs NumPy):     67.2x
Speedup (JIT vs NumPy):     31.8x
Speedup (JIT vs JAX):       0.5x

        __
        __