arrow
返回

MathOptAI.jl: Embed Trained Machine-Learning Predictors into JuMP Models

delete2026-04-01
delete0
PRE
AI
D
Dowson, Oscar *
P
Parker, Robert B.
B
Bent, Russell
DOI:10.1287/ijoc.2025.1446delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
我们介绍MathOptAI.jl,一个用于将训练好的机器学习预测器嵌入JuMP模型的开放源代码Julia库。MathOptAI.jl可以将多种神经网络、决策树和高斯过程嵌入到更大的数学优化模型中。MathOptAI.jl支持一系列基于Julia的机器学习库,如Lux.jl和Flux.jl。此外,MathOptAI.jl使用Julia的Python接口来提供对PyTorch模型的支持。当PyTorch支持与MathOptAI.jl的灰箱公式结合时,与PyTorch模型相关的函数、雅可比矩阵和海森矩阵的评估被卸载到Python中的GPU上,而其余的非线性算子则在Julia的CPU上评估。
Keyword:
optimization
machine learning
Julia
JuMP

期刊

I
INFORMS Journal on Computing
IF:
2.1
论文数:
90
被引数:
3.2K

机构

L
los alamos national laboratory
学者数:
845
论文数: 318
被引数: 0
U
united states department of energy (doe)
学者数:
11.3W
论文数: 9.6W
被引数: 246
引用论文

引用论文

暂无论文信息