🎉 We're happy to announce the PyEPO 1.0.0 release. 🎉
We are excited to announce the support of MPAX
, a PDHG-based optimization framework. With batched linear programming on either CPU or GPU, MPAX
leverages JIT
compilation for faster execution and enhanced scalability. Unlike traditional solvers, MPAX
can run entirely on the GPU, eliminating costly CPU-GPU communication overhead during training.
In general, MPAX
is particularly highly efficient for solving large-scale optimization problems. To see it in action, check out our Jupyter Notebook Tutorial.
Additional Updates in PyEPO 1.0.0
- Further Vectorization for Computation: Eliminated unnecessary
for-loops
, enhancing training efficiency. - Bug Fixes in Perturbation Algorithms: Resolved issues with solution caching in
perturbedOpt
,perturbedFenchelYoung
,implicitMLE
, andadaptiveImplicitMLE
.
We're eager for you to test these out and share your feedback with us. As always, thank you for being a part of our growing community!