Quickstart¶
Minimal recipes for each thing qpax can do. Every code block on the
following pages is the verbatim contents of a runnable script under
examples/,
so you can copy a file out of the repo and run it as-is. Pick the recipe
you need:
- Solve a QP — single forward solve.
- Differentiate a QP — pass gradients through the
solution with
jax.grad. - Batched solving — solve many independent QPs in
parallel with
jax.vmap. - Batched differentiating — gradients through a whole batch at once.