lyapunov exponents with JITCODE

Question

I am using JITCODE to calculate the lyapunov exponents of Lorenz oscillator.

Here is the simple script following the documentation:

import numpy as np