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>
AMICI-dev · Jan 30, 2025 · 64de63a · 64de63a
1 parent 923b8f4
commit 64de63a
Show file tree

Hide file tree

Showing 5 changed files with 646 additions and 272 deletions.
diff --git a/documentation/rtd_requirements.txt b/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