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

岗位职责

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

任职要求

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

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

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

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

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

6. 有密码算法经验者优先。

岗位标签

NEW

特色标签

AI算子工具栈、AMD ROCm、C++、GPU核函数、GPU编程框架、kernel、Linux开发/部署经验、NVIDIA CUDA、Python、单指令多数据、可复用算子库、向量化计算、定制化AI芯片、嵌入式软件经验、并行向量运算、并行计算平台、底层性能调优、开放GPU计算平台、异构计算平台、性能优化、数学算子、数据流水线、流式计算路径、算子实现、算子开发工具链、算子性能优化、算子构建工具链、算子集合、算子集合库、编译器开发经验、自主可控芯片、计算内核、计算数据流、运算符、领域专用芯片