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

基于自研计算芯片，研发高性能算子、对算子和应用进行调优；

针对业务落地过程中的性能与泛化性问题，不断迭代算子库；

能够编写高效的计算核和数据流，交付性能极致的算子；

不断迭代算子工具链，以确保其完整性和易用性，和业务一起构建生态。

硕士及以上学历，计算机/电子/通信等专业，4年以上相关工作经验，有AI算子开发经验者优先；

熟悉Linux开发环境，掌握Python/C++等语言， 具备良好的编程基础以及系统设计能力；

熟悉GPU或SIMD编程体系结构和原理，对Cuda、ROCm算子具备一定的开发能力；

熟悉计算机架构，善于分析关键路径； 加分项

有线性代数和概率论基础，熟悉数学建模、算法设计和优化方法者优先；

有密码算法经验者优先。