职位描述
- 构建昆仑芯监督学习、强化学习加速框架,支撑物理 AI 模型训练
- 开发自动驾驶、具身智能高性能计算库,优化核心算法与算子
- 完成主流模型在昆仑芯平台的训推部署与性能优化
- 支持重点客户项目,推动模型适配与业务落地。任职要求
- 相关专业硕士及以上学历,精通 C++、Python,基础扎实
- 熟悉 GPU 架构、CUDA 并行优化及 GDB、Nsight 等工具
- 熟悉 PyTorch、DeepSpeed、Megatron 及分布式训练;了解 PPO、GRPO 和主流 RL 框架
- 熟悉 VLA、WAM、3DGS、世界模型, 具备模型优化经验者优先