Notes JIT compilation witH JAX does not work for any of the methods due to a "ConcretizationTypeError"? scipy.integrate methods are not implemented in JAX yet, so get_arc_length is using regular scipy