Quantum optimal control based on dynamiqs, diffrax and JAX - View it on GitHub