jax vectorisation (#2636)

* fix vectorisation

* decorators, preequilibration

* add documentation

* fix petab_simulate

* fix notebook

* update/fix doc?

* fix rtd

* fix docstring

* Update python/sdist/amici/jax/petab.py

Co-authored-by: Daniel Weindl <dweindl@users.noreply.github.com>

---------

Co-authored-by: Daniel Weindl <dweindl@users.noreply.github.com>

Author: FFroehlich

documentation/rtd_requirements.txt

@@ -5,6 +5,7 @@ setuptools>=67.7.2
 pysb>=1.11.0
 jax>=0.4.26
 diffrax>=0.5.0
+interpax>=0.3.4
 matplotlib==3.7.1
 nbsphinx==0.9.1
 nbformat==5.8.0