用于学习神经算子的库
机器学习
2026-02-02 v5 人工智能
摘要
我们提出了 NeuralOperator，一个用于算子学习的开源 Python 库。神经算子将神经网络推广为函数空间之间的映射，而非有限维欧几里得空间之间的映射。它们可以在给定不同离散化下的输入和输出函数上进行训练和推断，并满足离散化收敛性质。作为 PyTorch 官方生态系统的一部分，NeuralOperator 在一个高质量、经过测试的开源软件包中提供了训练和部署神经算子模型以及开发新模型所需的所有工具。它将前沿模型和可定制性与平缓的学习曲线和面向新手的简单用户界面结合在一起。
引用
@article{arxiv.2412.10354,
title = {A Library for Learning Neural Operators},
author = {Jean Kossaifi and Nikola Kovachki and Zongyi Li and David Pitt and Miguel Liu-Schiaffini and Robert Joseph George and Boris Bonev and Kamyar Azizzadenesheli and Julius Berner and Valentin Duruisseaux and Anima Anandkumar},
journal= {arXiv preprint arXiv:2412.10354},
year = {2026}
}