Use pure-JAX minimizers
Some supported in JAX - but not with arbitrary constraints.
Would allow to use pmap
/vmap
on fits, run fits in parallel on toys or for pulls and impacts.
Some supported in JAX - but not with arbitrary constraints.
Would allow to use pmap
/vmap
on fits, run fits in parallel on toys or for pulls and impacts.