岗位描述(节选)
岗位职责
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……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。