岗位描述
职位描述:
你是否希望让先进 AI 模型训练得更快、更稳、更高效,并以系统能力支持持续的算法创新?
聚宽投资正在寻找兼具算法理解与系统能力的 AI 研究员。你将围绕大规模训练、推理优化和研究基础设施开展工作,提升模型研发效率和实验可靠性。
岗位职责:
研究大规模模型训练、分布式计算、并行策略和训练稳定性;
探索模型压缩、低精度训练、推理加速和资源效率优化;
建设高效、可复现、可观测的训练与实验基础设施;
分析算法与软硬件系统瓶颈,提出端到端优化方案;
与模型研究团队协作,加速前沿算法的验证和迭代。
任职要求:
海内外知名高校本科及以上学历,计算机、软件工程、电子信息等相关专业优先;
熟悉 PyTorch 及主流深度学习训练框架;
熟悉分布式训练、GPU 计算、性能分析或推理优化中的至少一项;
具备扎实的工程能力和系统问题定位能力;
对算法与系统协同优化具有浓厚兴趣。
加分项:
熟悉 CUDA、Triton、NCCL、DeepSpeed、Megatron 或 Ray;
有大模型预训练、集群调度、训练容错或高性能数据管线经验;
有模型量化、剪枝、蒸馏或编译优化经验;
有高质量开源项目、系统论文或大规模生产实践经验。