AI框架研发工程师（TensorFlow/PyTorch）
岗位摘要
优化TensorFlow框架下各类神经网络的性能，包括网络性能瓶颈分析、框架内部开发及性能优化；紧跟深度学习社区最新技术进展，及时将前沿性能优化技术迁移到MLU平台；保持技术竞争力与创新性，推动平台整体性能提升
岗位职责
- 优化TensorFlow框架下各类神经网络的性能，包括网络性能瓶颈分析、框架内部开发及性能优化
- 紧跟深度学习社区最新技术进展，及时将前沿性能优化技术迁移到MLU平台
- 保持技术竞争力与创新性，推动平台整体性能提升
任职要求
- 具备扎实的C++和Python编程能力，拥有良好的数据结构与算法基础
- 学习能力强，具有强烈的求知欲和自我驱动能力，具备良好的沟通能力和团队协作精神
- 【加分项】熟悉TensorFlow等主流深度学习框架内部模块与机制，对异步执行器、内存分配器、设备运行时接口及图优化器有深度了解和实际开发经验，了解PyTorch、PaddlePaddle等框架亦可
- 【加分项】熟悉XLA、MLIR等深度学习编译技术
- 【加分项】在搜索、广告、推荐领域相关模型优化及技术发展方面有一定实际经验
- 学历要求：硕士及以上
加分项
- 1：熟悉TensorFlow等主流深度学习框架，熟悉框架内部的模块、机制
- 对tensorflow异步执行器（executor）、内存分配器（如bfc allocator）、设备运⾏时接口（如cuda runtime）、图优化器（grappler）有深度的了解和实际的开发经验
- 如果了解其他框架，如：pytorch、paddle，亦可
- 2：熟悉XLA、MLIR深度学习编译技术
- 3：对搜索广告推荐领域相关模型优化、技术发展有一定的实际经验
- 芦先生 刚刚活跃
福利待遇
- 薪资范围：30K–60K，共15薪
- 工作地点：北京市海淀区北航致真大厦D座
工作地址
北京海淀区北航致真大厦D座