Reduce NonSequential JIT compile times via jax.lax.scan and optimizer state packing#5

Copilot wants to merge 2 commits intomainParametric-Matrix-Models/pyPMM:mainfrom

copilot/reduce-nonsequential-compile-timeParametric-Matrix-Models/pyPMM:copilot/reduce-nonsequential-compile-timeCopy head branch name to clipboard