





















Abstract:We present \texttt{this http URL}, an open-source Julia library for embedding trained machine learning predictors into a JuMP model. \texttt{this http URL} can embed a wide variety of neural networks, decision trees, and Gaussian Processes into a larger mathematical optimization model. In addition to interfacing a range of Julia-based machine learning libraries such as \texttt{this http URL} and \texttt{this http URL}, \texttt{this http URL} uses Julia's Python interface to provide support for PyTorch models. When the PyTorch support is combined with \texttt{this http URL}'s gray-box formulation, the function, Jacobian, and Hessian evaluations associated with the PyTorch model are offloaded to the GPU in Python, while the rest of the nonlinear oracles are evaluated on the CPU in Julia. \MathOptAI is available at this https URL under a BSD-3 license.
| Subjects: | Machine Learning (cs.LG); Optimization and Control (math.OC) |
| MSC classes: | 90-04 |
| Cite as: | arXiv:2507.03159 [cs.LG] |
| (or arXiv:2507.03159v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2507.03159 arXiv-issued DOI via DataCite |
From: Oscar Dowson [view email]
[v1]
Thu, 3 Jul 2025 20:32:08 UTC (35 KB)
[v2]
Mon, 25 May 2026 02:45:33 UTC (42 KB)
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。