深度学习框架图编译工程师
岗位摘要
优化Torch Compile模式下各类神经网络的整体性能，涵盖网络性能瓶颈分析、编译耗时优化及codegen性能优化等方向；紧跟PyTorch社区最新技术进展，及时将前沿性能优化技术迁移到MLU平台，保持技术竞争力和创新性
岗位职责
- 优化Torch Compile模式下各类神经网络的整体性能，涵盖网络性能瓶颈分析、编译耗时优化及codegen性能优化等方向
- 紧跟PyTorch社区最新技术进展，及时将前沿性能优化技术迁移到MLU平台，保持技术竞争力和创新性
任职要求
- 具备较强的C++和Python编程能力，拥有扎实的数据结构与算法基础
- 学习能力强，具有强烈的求知欲和自我驱动能力，具备良好的沟通能力和团队协作精神
- （加分项）熟悉PyTorch、TensorFlow等主流深度学习框架，了解框架内部模块与机制
- （加分项）熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术
加分项
- 1：熟悉PyTorch/TensorFlow等主流深度学习框架，熟悉框架内部的模块、机制
- 2：熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术
福利待遇
- 薪资范围25K至50K，共15薪，薪酬待遇优厚
工作地址
深圳南山区田厦金牛广场3406