岗位职责
- 基于自研计算芯片,研发高性能算子、对算子和应用进行调优;
- 针对业务落地过程中的性能与泛化性问题,不断迭代算子库;
- 能够编写高效的计算核和数据流,交付性能极致的算子;
- 不断迭代算子工具链,以确保其完整性和易用性,和业务一起构建生态。
职位要求
- 硕士及以上学历,计算机/电子/通信等专业,4年以上相关工作经验,有AI算子开发经验者优先;
- 熟悉Linux开发环境,掌握Python/C++等语言, 具备良好的编程基础以及系统设计能力;
- 熟悉GPU或SIMD编程体系结构和原理,对Cuda、ROCm算子具备一定的开发能力;
- 熟悉计算机架构,善于分析关键路径; 加分项
- 有线性代数和概率论基础,熟悉数学建模、算法设计和优化方法者优先;
- 有密码算法经验者优先。