训练框架研发工程师
岗位摘要
参与公司训练底层框架研发，负责超大规模训练技术在实际业务场景中的计算效率优化；在超大规模训练场景下，探索前沿的分布式系统设计，包括通信优化、内存优化、计算优化等；参与建设PyTorch 2.0/TorchDynamo/TorchInductor/Triton等编译技术优化
岗位职责
- 参与公司训练底层框架研发，负责超大规模训练技术在实际业务场景中的计算效率优化
- 在超大规模训练场景下，探索前沿的分布式系统设计，包括通信优化、内存优化、计算优化等
- 参与建设PyTorch 2.0/TorchDynamo/TorchInductor/Triton等编译技术优化
任职要求
- 扎实的系统功底和编程能力，熟悉C++/Python，熟悉常见数据结构和设计模式
- 熟悉深度学习基本原理和AI框架（TensorFlow/PyTorch/JAX等），有PyTorch二次开发经验者优先
- 熟悉深度学习框架优化/问题定位相关工具链基本原理，有二次开发经验者优先
- 拥有高性能计算或模型优化经验和兴趣（CUDA/Triton/OpenMP）
- 热爱技术，有较强的学习能力、复杂问题归纳梳理能力
工作地址
北京海淀区蓟门壹号6楼8F