职位描述
1、结合前沿业务场景,构建昆仑芯AI大规模训练推理系统
2、负责大模型分布式训练、推理框架的适配与调优,设计千卡级集群通信加速、混合精度训练等方案
3、为昆仑芯AI芯片各系列高性能加速芯片提供软件栈,包括框架,图编译器以及周边产品的技术落地
4、AI芯片性能深度学习高性能计算库开发,支持各种AI场景,持续提升系统效能任职要求
1、2027届及以后在校生(微电子,计算机,数学等相关专业)
2、熟悉Python/C/C++,良好的编程习惯,熟悉计算机基础算法和数据结构,熟悉 Linux 开发环境
3、熟悉深度学习相关知识,有框架(Pytorch, PaddlePaddle,TVM,MLIR,ONNX等)开发或使用经验优先
4、熟悉CUDA,OpenCL等高性能计算编程经验优先