Performance examples¶
These scripts highlight performance/scaling and “why JAX?”.
JAX vs NumPy: Biot–Savart timing¶
Script:
examples/performance/jax_vs_numpy_biot_savart_speed.py
This compares:
a simple NumPy baseline that loops over evaluation points, and
the vectorized JAX implementation (compiled with
jax.jit).
Run:
python examples/performance/jax_vs_numpy_biot_savart_speed.py --n-eval 512 --repeat 20