【[104星]LQRax:基于JAX的GPU友好、自动可微分的连续时间LQR求解器。亮点:1. 利用JAX的scan机制加速数值模拟;2. 支持单Agent和多Agent非线性控制算法的快速原型设计;3. 借助JAX的vmap机制实现GPU上的大规模最优控制计算】
'LQRax: GPU-friendly, auto-differentiable solver for continuous-time LQR problems based on Riccati equations, enabled by JAX.'
GitHub: github.com/MaxMSun/lqrax
轨迹优化 最优控制 LQR求解器 人工智能 ai兴趣创作计划