蚂蚁集团-算子开发工程师-上海/北京

岗位描述
1、基于自研计算芯片，研发高性能算子、对算子和应用进行调优；
2、针对业务落地过程中的性能与泛化性问题，不断迭代算子库；
3、能够编写高效的计算核和数据流，交付性能极致的算子；
4、不断迭代算子工具链，以确保其完整性和易用性，和业务一起构建生态。

岗位要求
1、硕士及以上学历，计算机/电子/通信等专业，4年以上相关工作经验，有AI算子开发经验者优先；
1、熟悉Linux开发环境，掌握Python/C++等语言， 具备良好的编程基础以及系统设计能力；
2、熟悉GPU或SIMD编程体系结构和原理，对Cuda、ROCm算子具备一定的开发能力；
3、熟悉计算机架构，善于分析关键路径；
加分项
1.有线性代数和概率论基础，熟悉数学建模、算法设计和优化方法者优先；
2.有密码算法经验者优先。