jax vectorisation

[FFroehlich]

Looks like some recent jax update already brought massive improvements to jit time. This reduces compile time by about 4x, (~28s to ~6s on M2 MBA for Bachmann MSB2011)

[codecov]

Codecov Report

Attention: Patch coverage is 96.77419% with 2 lines in your changes missing coverage. Please review.

Project coverage is 77.09%. Comparing base (923b8f4) to head (981cefa).
Report is 1 commits behind head on develop.

Files with missing lines	Patch %	Lines
python/sdist/amici/jax/petab.py	96.29%	2 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #2636      +/-   ##
===========================================
+ Coverage    77.02%   77.09%   +0.07%     
===========================================
  Files          332      332              
  Lines        22972    23012      +40     
  Branches      1478     1478              
===========================================
+ Hits         17694    17742      +48     
+ Misses        5267     5259       -8     
  Partials        11       11              

Flag	Coverage Δ
cpp	74.20% <83.87%> (+0.07%)	⬆️
cpp_python	33.41% <0.00%> (-0.07%)	⬇️
petab	39.00% <93.54%> (+0.13%)	⬆️
python	72.18% <85.48%> (+0.07%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Files with missing lines	Coverage Δ
python/sdist/amici/jax/model.py	87.74% <100.00%> (+0.57%)	⬆️
python/sdist/amici/jax/petab.py	97.42% <96.29%> (-0.72%)	⬇️

... and 3 files with indirect coverage changes

[dweindl]

Looks good to me.

[sonarqubecloud]

Quality Gate passed

Issues
0 New issues
0 Accepted issues

Measures
0 Security Hotspots
0.0% Coverage on New Code
0.0% Duplication on New Code

See analysis details on SonarQube Cloud