[General] GPU Accelerated Trajectory Optimization with External Transcription

未关闭
#676 6 条评论 0 个 reaction 已指派 1 人 在 GitHub 查看

维护者通常 1 天内回复

@ocots 已经在做这个了。

开始于 2025年11月15日。

评估

这个 Issue 还没有评估数据。

描述

@jbcaillau and @ocots,

I am trying to figure out the best way to implement optimal control problems for trajectory optimization in Julia to be solved with GPUs, making the best use of existing packages.

After chatting with @amontoison, I believe that OptimalControl.jl can be a useful framework to communicate with the appropriate modeling tools (e.g., Examodels) and solvers (e.g., those in MadSuit), besides handling the integrator part.

Unfortunately, I haven’t been able to completely derive the dynamics of my quadruped robot from scratch, which I believe is necessary for OptimalControl.jl.

Taking inspiration from TORA.jl (cc @ferrolho), I thought of using RigidBodyDynamics.jl (RBD) to derive them from a URDF file. In TORA.jl, this approach calls the RBD dynamics inference API compatible with AD to calculate both the Jacobian and Hessian. From my chat with @amontoison, I believe this could be made compatible with OptimalControl.jl by using the NLPModels Manual API: https://jso.dev/tutorials/create-a-manual-model/. The issue here is that this approach would require me to extend and ensure that RigidBodyDynamics.jl can perform dynamics inference on the GPU, as far as I know (AFAIK).

A while back, I also tried to rewrite RigidBodyDynamics.jl to directly create the appropriate constraints in a given JuMP model instead of solving the dynamics (https://github.com/andrewrosemberg/RigidBodyDynamics.jl/pull/1). However, it seems that my naive implementation produces extremely large expressions, causing JuMP to stall when trying to check if they belong to the model. I now believe the rewrite would need to be much more intricate, ensuring that the expressions are created cleverly, avoiding redundancy, and being appropriately compatible with how the GPU kernels should be created for good performance.

Sorry for the long post, but I wanted to see what you and others from the community think. :)

主要语言
Julia
星标
135
派生
11
平均合并
7 小时 49 分钟
30 天内合并 PR
23

环境准备

  • 没有 Dockerfile 或 Docker Compose 文件
  • 没有 Pull Request 模板
  • 阅读贡献指南

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

control-toolbox/OptimalControl.jl 的其他 Issue

查看 control-toolbox/OptimalControl.jl 的全部 Issue

相似的 Issue

更多 Julia Issue

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。