训练框架研发工程师-2026届

薪资面议
人工智能
上海
不限

投递时间:2025年12月23日-2029年1月27日

岗位职责
1. 参与公司训练底层框架研发, 负责超大规模训练技术在公司实际业务场景中的计算效率优化. 2. 超大规模训练场景下,探索前沿的分布式系统设计,包括不限于通信优化、内存优化、计算优化等。 3. 参与建设 PyTorch 2.0/TorchDynamo/TorchInductor/Triton 等编译技术优化.
岗位要求
1. 扎实系统功底和编程能力,熟悉 C++/Python,熟悉常见数据结构和设计模式; 2. 熟悉深度学习基本原理和 AI 框架 (TensorFlow/PyTorch/JAX等), 有对 Pytorch 二次开发经验者优先 3. 熟悉深度学习框架优化/问题定位相关工具链基本原理, 有二次开发经验者优先 4. 拥有高性能计算或模型优化经验和兴趣 (CUDA/Triton/OpenMP); 5. 热爱技术、有较强的学习能力、复杂问题归纳梳理能力.
新漕河泾商务中心B座25层
牛客安全提示:如发现虚假招聘、广告,或以任何名义索要证件、费用,或诱导异地入职、参与培训等均属违法行为,发现请 立即举报