[FR] iminuit + jax benchmark

ComPWA / benchmarks

Collection of scripts and notebooks for benchmarking ComPWA workflows

https://compwa.github.io/benchmarks

1 stars 0 forks source link

[FR] iminuit + jax benchmark #16

Open mmikhasenko opened 1 year ago

mmikhasenko commented 1 year ago

I notice interesting studies in the iminuit package. Just dropping the links here, might be linked somewhere in README, or used for an inspiration when writing future benchmarks.

RooFit vs numba JIT https://iminuit.readthedocs.io/en/latest/benchmark.html
Jax JIT wiwth gradient is doing slightly worse than numba JIT without grad https://iminuit.readthedocs.io/en/latest/notebooks/automatic_differentiation.html#Fit-of-a-gaussian-model-to-a-histogram

redeboer commented 1 year ago

Nice, thanks! Indeed good to keep in mind for general fitting benchmarks

Some remarks with regard to TW.

Numba is not yet working too well as backend in TW. Probably doing something wrong there with JIT
For autodiff with large amplitude models, there's some work going on on this here: https://github.com/ComPWA/compwa-org/pull/195