岗位职责：

1.负责AI大模型算子库开发及优化；

2.与芯片设计团队密切配合，分析网络性能瓶颈，提出优化策略，达成高性能计算加速、编译优化、链路优化；

3.结合算法发展趋势，提出芯片架构优化建议，完善硬件架构；

4.支撑大模型推理及微调业务的运行加速与优化；

5.将PyTorch、vLLM、DeepSpeed和Megatron等深度学习框架适配到公司自研的芯片上。

任职要求：

1.计算机科学、电子工程或相关领域的硕士学位；

2.熟悉C/C++/Python软件编程语言；

3.深入理解计算机数据结构和算法设计，熟悉指令流水、计算机体系结构；

4.了解深度学习算法，了解主流深度学习模型，对主流深度学习框架；Pytorch/Tensorflow/Padddle有一定的应用或开发经验；

5.熟悉并行/异构计算体系结构，熟悉CUDA/HIP/OpenCL等异构编程语言或cutlass等算子开发框架，在GPU/TPU/NPU一种平台有相关开发和优化经验，或有高性能库cuBlas/cuDNN/MIOpen开发和性能优化经验；

6.优秀的分析问题和解决问题的能力，良好的团队沟通能力、技术攻关能力，责任心强，具备一定抗压能力。