具身Infra工程师
研发中心 · 上海市 / 北京市 / 广东省·深圳市 · 社招
AI 摘要智能生成
再搭起云端-边端协同架构,支撑复杂任务拆解与大模型委派执行
职位描述
1、为 VLA、Word Action Model 等多模态大模型的训练扫清效率障碍:在异构集群上压低长序列 Transformer 与 Diffusion Policy 的计算、通信开销; 2、打造 GPU 并行仿真与策略推理一体的 RL 训练框架(Isaac Sim、MuJoCo MJX、Enigma 等),打通 Sim2Real 训练回路,落地域随机化与真实数据混合回放,让 PPO、SAC、TD-MPC 的采样效率和每秒交互步数(SPS)持续爬坡; 3、把 VLA 跑进机器人端侧(Orin、x86 实时系统):量化、稀疏化、算子融合多管齐下,压住流式推理延迟,在速度与成功率之间找到最优平衡;再搭起云端-边端协同架构,支撑复杂任务拆解与大模型委派执行。
任职要求
- 本科以上,计算机、自动化、机器人等专业,3 年以上 AI Infra 或高性能计算经验;
- Python 与 C++ 是你的主力语言,写过 CUDA、调过性能;Megatron、DeepSpeed 用得熟;
- 懂 PPO 这类主流强化学习算法,RLLib、SampleFactory、Isaac Gym 至少上手过一个;了解 Isaac Sim、MuJoCo、Gazebo,能独立搭起并行仿真训练闭环;
- TensorRT、TVM、OpenXLA 至少精通其一,有端侧部署落地案例。