2 ms·

Jax is ceres but for python!