AI应用工程师

岗位职责

1、基于寒武纪软硬件产品，完成AI推理/AI训练相关适配和定制化开发工作；

2、完成大模型的训推移植、部署和优化，以及模型相关的性能/精度分析、调优。

任职要求

1、本科及以上学历，计算机、微电子、通信、自动化、应用数学、物理等相关专业；

2、熟练掌握C++/Python/Shell等编程语言，掌握Linux操作系统下CMake、GDB、Git、Docker等常用开发和调试工具；

3、熟悉PyTorch或TensorFlow等深度学习框架，熟悉torch.compile或XLA优化路径，有框架深度应用和二次开发经验优先；

4、熟悉大模型推理框架和开源库，如vLLM、SGLang、Diffusers等，有相关大规模应用部署、二次开发、性能优化经验者优先；

5、熟悉大模型训练框架和开源库，如Transformers、DeepSpeed、Megatron等，有相关大规模SFT/LoRA训练和性能、精度调优经验者优先。