深度学习框架图编译工程师
岗位摘要
优化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等深度学习编译技术
- 胡先生 刚刚活跃
工作地址
北京海淀区北航致真大厦D座16层