MelonOpt

A modular toolkit for large-scale robust and stochastic optimization, with first-class support for learning pipelines. Problems are composed from reusable building blocks and differentiated end to end on top of JAX.

The flagship project of the group's open-source effort, with extensive docs, tutorials, and a growing collection of solver backends.