If you are not redirected automatically, go to https://qiskit-community.github.io/qiskit-dynamics/stable/0.4/userguide/how_to_use_jax.