一个将普通 JavaScript 控制流编译为可微分神经网络的研究项目,权重手工写入而非训练得到。
pnnl/neuromancer
基于 PyTorch 的可微编程库,用于参数化约束优化、物理信息系统辨识和参数化模型预测控制。
- Star
- 1,383
- Fork
- 185
- 未关闭 Issue
- 16
- 最近提交
- 4小时前
这是什么?
NeuroMANCER 是一个基于 PyTorch 的可微编程库,用于学习参数化约束优化问题的解、构建融合物理知识的动态系统模型,以及学习控制策略。用户可以把神经网络与科学计算结合起来,将领域方程、约束和先验知识直接嵌入训练过程。
库提供符号化编程接口,用于定义变量、约束和目标函数,并通过 Node 和 System 等抽象组合神经模块与动力学模型。支持的方法包括神经 ODE、Koopman 算子、SINDy、Kolmogorov-Arnold 网络、函数编码器,以及 DeepONet、傅里叶神经算子等神经算子,还可用于求解 PDE 的物理信息神经网络。控制方面包含可微预测控制(DPC)和用于安全约束学习的可微凸优化层。
仓库附带大量可直接在 Colab 中运行的教程和领域示例,涵盖建筑热动力学与暖通空调控制等场景,并与 PyTorch Lightning 集成以支持 GPU 和多卡训练。项目可通过 pip 安装。
核心功能
- 01符号化接口,可将物理方程、约束和目标函数嵌入学习问题。
- 02通过基于罚函数的损失和可微凸优化层求解参数化约束优化问题。
- 03使用神经 ODE、Koopman 算子、SINDy 和状态空间模型学习动力系统模型。
- 04通过可微预测控制(DPC)学习非线性系统的控制策略。
- 05提供物理信息神经网络以及 DeepONet、傅里叶神经算子等神经算子。
- 06集成 PyTorch Lightning,支持 GPU 和多卡训练。
适用场景
- 建筑能源管理,例如用 HVAC 系统学习室内温度控制。
- 基于实测数据学习物理系统的代理模型,同时满足已知方程约束。
- 在大量参数取值下快速求解同类约束优化问题。
- 为扩散方程、流体等偏微分方程原型化物理信息神经网络。
快速上手
- 1
从 PyPI 安装该库
pip install neuromancer - 2
在 Python 中运行README中的约束优化示例(定义模型、节点和损失函数)
使用前须知
- 许可证元数据显示为 NOASSERTION,使用前需在代码仓库中核实具体条款。
- 大多数教程和示例基于 PyTorch,在没有 GPU 的 CPU 环境下运行可能较慢。
标签
常见问题
- neuromancer 是什么?
- 基于 PyTorch 的可微编程库,用于参数化约束优化、物理信息系统辨识和参数化模型预测控制。 NeuroMANCER 是一个基于 PyTorch 的可微编程库,用于学习参数化约束优化问题的解、构建融合物理知识的动态系统模型,以及学习控制策略。用户可以把神经网络与科学计算结合起来,将领域方程、约束和先验知识直接嵌入训练过程。
- neuromancer 可以免费使用吗?用的什么许可证?
- neuromancer 以 Other 许可证开源。用于商业产品前,请先确认许可证条款。
- neuromancer 还在维护吗?
- 活跃维护中。neuromancer 最近一次提交是在 2026年10月22日。
- neuromancer 适合用来做什么?
- 建筑能源管理,例如用 HVAC 系统学习室内温度控制。 基于实测数据学习物理系统的代理模型,同时满足已知方程约束。 在大量参数取值下快速求解同类约束优化问题。 为扩散方程、流体等偏微分方程原型化物理信息神经网络。
- neuromancer 怎么安装和运行?
- 从 PyPI 安装该库: pip install neuromancer 在 Python 中运行README中的约束优化示例(定义模型、节点和损失函数)
- 在哪里获取 neuromancer?
- 源代码和安装说明在 GitHub 上:https://github.com/pnnl/neuromancer
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
面向文本、视觉、音频和多模态的前沿机器学习模型定义框架,可用于推理和训练。
Keras 3 是一个多后端深度学习框架,可运行于 JAX、TensorFlow、PyTorch 之上,OpenVINO 仅用于推理。
Ultralytics YOLO 是一个 Python 计算机视觉库,支持目标检测、分割、分类、姿态估计和目标跟踪等任务。
一个极简、易于修改的 LLM 训练框架,覆盖分词、预训练、微调、评估到对话推理全流程,可在单台 GPU 节点上运行。
基于 PyTorch 的计算机视觉工具库,支持目标检测、实例分割和图像分类的训练与模型导出。