tiny-differentiable-simulator

GitHub
1.4k 137 较难 1 次阅读 1周前Apache-2.0开发框架其他
AI 解读 由 AI 自动生成,仅供参考

tiny-differentiable-simulator 是一款专为强化学习与机器人领域设计的轻量级物理仿真库。它采用纯头文件 C++ 与 CUDA 编写,无需任何外部依赖即可运行,极大地降低了部署门槛。

该工具主要解决了传统物理引擎难以直接融入深度学习训练流程的痛点。通过原生支持前向与反向模式自动微分,它能将刚体动力学、接触模型及各类执行器(如电机、串联弹性驱动器)的计算过程转化为可导操作,让研究人员能够轻松结合神经网络进行端到端的策略优化与参数推断。

tiny-differentiable-simulator 特别适合从事机器人控制算法研究、强化学习开发以及需要高保真物理仿真的科研人员与工程师使用。其核心技术亮点在于高度的模板化设计:用户不仅可以使用常规的浮点数进行高速计算,还能无缝切换至 CppAD、Stan Math 等自动微分标量类型,甚至利用定点整数数学实现跨平台的确定性计算。此外,凭借高效的 CUDA 并行加速能力,它能在单张 RTX 2080 显卡上以每秒 50 帧的速度并行运行数千个仿真环境,显著提升了大规模训练的效率。配合内置的 OpenGL 或基于 Web 的 MeshCat 可视化方案,开发者可以直观地调试与展示仿真结果。

使用场景

某机器人研发团队正在为四足机器狗开发基于强化学习的复杂地形适应算法,需要高频次训练策略网络以应对真实世界的物理不确定性。

没有 tiny-differentiable-simulator 时

  • 梯度获取困难:传统物理引擎(如 Bullet 或 MuJoCo)不支持自动微分,团队必须使用样本效率极低的黑盒优化算法,或通过有限差分法近似梯度,导致训练收敛缓慢且不稳定。
  • 仿真速度瓶颈:在单张 RTX 2080 显卡上难以并行运行数千个环境实例,数据采集速度慢,无法在合理时间内完成大规模强化学习所需的亿级步数训练。
  • 部署依赖沉重:现有方案往往依赖庞大的外部库和复杂的运行时环境,使得将训练好的策略迁移到嵌入式边缘设备时面临严重的兼容性和体积问题。
  • 模拟与现实差距大:由于缺乏可微分的接触模型和精确的驱动器动力学(如 SEA),仿真中的摩擦和弹性行为与真实硬件偏差较大,导致“虚实迁移”失败率高。

使用 tiny-differentiable-simulator 后

  • 端到端梯度传播:利用其头文件模板化设计,团队直接结合 CppAD 或 Stan Math 实现前向与反向模式自动微分,精准计算物理状态对控制参数的梯度,大幅提升了策略网络的收敛速度。
  • GPU 大规模并行:借助原生 CUDA 支持,在单卡上即可同时运行数千个仿真实例并保持 50 FPS 高帧率,将数据采集效率提升了两个数量级,缩短了研发周期。
  • 零依赖轻量部署:作为无外部依赖的纯头文件 C++ 库,代码可直接嵌入资源受限的机器人控制器中,实现了从训练到部署的无缝衔接,且保证了跨平台的确定性计算。
  • 高保真物理建模:通过内置的脉冲级 LCP 接触模型及串联弹性驱动器(SEA)动力学,仿真结果高度逼近真实物理特性,显著缩小了 Sim2Real 差距,让机器狗在真机上一次成功。

tiny-differentiable-simulator 通过提供高性能、可微分且零依赖的物理仿真内核,彻底解决了机器人强化学习中梯度难算、仿真慢及虚实迁移难的三大核心痛点。

运行环境要求

操作系统
  • Linux
  • macOS
  • Windows
GPU
  • 可选
  • 若需并行加速则必需 NVIDIA GPU(示例提及 RTX 2080),需支持 CUDA
  • CPU 模式下无需 GPU
内存

未说明

依赖
notes该库为纯头文件库,零外部依赖。代码已模板化,支持自动微分标量类型(如 CppAD, Stan Math, ceres::Jet)或普通浮点数。若使用 MeshCat 可视化,需安装 Python、pip 和 meshcat,并运行 meshcat-server,推荐使用 Chrome 浏览器。固定点整数数学运算可提供跨平台确定性计算。
python未说明(仅在使用 MeshCat 可视化或运行相关示例时需要安装 Python)
CMake
C++17 兼容编译器
CUDA (可选,用于 GPU 加速)
TinyXML2 (内置)
MeshCat (可选,用于 Web 可视化)
pyzmq (可选,用于 MeshCat C++ 接口)
tiny-differentiable-simulator hero image

快速开始

微型可微分物理引擎

微型可微分物理引擎是一个无依赖的、仅包含头文件的 C++(及 CUDA)物理库。

  • 注意:该主仓库已从 google-research 迁移到 Erwin Coumans

目前实现了多种刚体动力学算法,包括正向和逆向动力学,以及基于冲量级线性互补问题(LCP)和基于力的非线性弹簧阻尼器的接触模型。此外还实现了电机、伺服电机和串联弹性执行器(SEA)动力学的执行器模型。

整个代码库采用模板化设计,因此可以使用前向和反向模式的自动微分标量类型,例如 CppAD、Stan Math fvar 和 ceres::Jet。该库也可以与常规的 float 或 double 精度值一起使用。另一种选择是使用内置的定点整数数学,它能够提供跨平台的确定性计算。

TDS 在单个 RTX 2080 CUDA GPU 上以每秒 50 帧的速度并行运行数千次仿真:

https://user-images.githubusercontent.com/725468/135697035-7df34b85-c73e-4739-9a76-dc114ce84c4c.mp4

提供了多种可视化工具,详见下文。

Bibtex 引用

请使用以下引用文献来标注本研究:

@inproceedings{heiden2021neuralsim,
  author =	  {Heiden, Eric and Millard, David and Coumans, Erwin and Sheng, Yizhou and Sukhatme, Gaurav S},
  year =		  {2021},
  title =		  {Neural{S}im: Augmenting Differentiable Simulators with Neural Networks},
  booktitle = {Proceedings of the IEEE International Conference on Robotics and Automation (ICRA)},
  url =		    {https://github.com/google-research/tiny-differentiable-simulator}
}

相关论文

  • “从 RGBD 视频中推断铰接式刚体动力学”(IROS 2022 提交)Eric Heiden、Ziang Liu、Vibhav Vineet、Erwin Coumans、Gaurav S. Sukhatme 项目页面
  • “NeuralSim:用神经网络增强可微分模拟器”(ICRA 2021)Eric Heiden、David Millard、Erwin Coumans、Yizhou Sheng、Gaurav S. Sukhatme。Arxiv 上的 PDF
  • “用神经网络增强可微分模拟器以缩小模拟到现实的差距”(RSS 2020 模拟到现实研讨会),Eric Heiden、David Millard、Erwin Coumans、Gaurav Sukhatme。Arxiv 上的 PDF视频
  • “交互式可微分仿真”,2020 年,Eric Heiden、David Millard、Hejia Zhang、Gaurav S. Sukhatme。Arxiv 上的 PDF

其他研究者使用 TDS 的相关研究

  • “铰接体的高效可微分仿真”(ICML 2021)Yi-Ling Qiao、Junbang Liang、Vladlen Koltun、Ming C. Lin Arxiv 上的 PDF

快速入门

开源版本使用 CMake 构建,需要支持 C++17 的编译器。

mkdir build
cd build
cmake ..
make -j

示例

对于可视化,支持两种方式:

OpenGL 3+ 可视化

  • tiny_opengl3_app,一个 OpenGL3 可视化工具

该可视化工具是本库的一部分,位于 src/visualizer/opengl 目录下。

MeshCat 可视化

  • MeshCat,一个基于 WebGL 的 Web 可视化工具

提供了 C++ ZMQ 接口。

在运行示例之前,请先安装 Python、pip 和 meshcat,启动 meshcat-server,并打开浏览器(推荐使用 Chrome 以获得更好的 three.js 体验)。

pip install meshcat
meshcat-server --open
这将打开 Chrome 浏览器,访问 http://localhost:7000/static/
然后编译并运行 tiny_urdf_parser_meshcat_example,使用优化/发布版本进行构建。

可以使用基于 TinyXML2 的解析器加载 URDF 文件。

MeshCat 可视化的所有依赖项均已包含在 third_party 目录中。


免责声明:这不是 Google 的官方产品。

版本历史

cuda_laikago_codegen22020/12/18
cuda_laikago_codegen2020/12/19

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|2天前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|3天前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

144.7k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

107.9k|★★☆☆☆|2天前
开发框架图像Agent

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|昨天
插件开发框架

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|2天前
语言模型图像Agent