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