深度学习开发实习生

1、使⽤类CUDA语言和国产AI芯片平台，开发深度学习高性能算⼦库的算⼦，⽐如激活算⼦、卷积算⼦、矩阵乘法、FlashAttention等。

2、对算⼦的正确性和精度进⾏对齐，对性能进行分析和极致调优。

3、支持算子在pytorch/vllm等深度学习框架上的高效执行。

1、本科及以上学历，计算机科学与技术、软件工程、人工智能等相关专业方向的在校生；

2、有良好的python、C/C++语⾔编程能力和动手实践能力，在算法基础，编译/运行时原理、体系结构等方面具有扎实的基础

3、了解GPU/NPU等异构加速卡的体系架构，AI软件栈组成等基本概念，有CUDA算子开发经验者优先

4、对深度学习有一定了解，有国产AI芯片使用经验者优先