返回
MathOptAI.jl: Embed Trained Machine-Learning Predictors into JuMP Models
DOI:10.1287/ijoc.2025.1446.png)
摘要
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
IF:
2.1
论文数:
90
被引数:
3.2K
机构
引用论文
暂无论文信息

