职位详情
AI框架研发工程师
面议
融通人力资源开发有限公司
无锡
3-5年
本科
06-19
工作地址

江南计算技术研究所山水东路188号

职位描述
(此岗位为外包岗位)
岗位职责:
1.熟悉业界主流框架的部署与集成,完成在自研异构芯片上的接入调试;
2.负责自研AI框架(含训练/推理引擎)的架构设计、核心模块开发及性能调优;
3.熟悉分布式并行基础原理,配合软硬件进行分布式训练推理策略的制定;
4.主导AI框架与自研芯片(GPU/NPU等)的软硬件协同设计,通过编译器优化(如图优化、算子融合、内存调度)提升计算效率。
岗位要求:
1.具备框架开发相关工作经验,从事过AI框架研发项目者优先;
2.深入了解主流大模型知识以及分布式训推技术((Megatron-LM、DeepSpeed));
3.了解AI编译器技术栈(MLIR/LLVM),掌握图优化、算子融合、多面体优化等编译加速方法优先;
4.有异构芯片(GPU/NPU)开发经验:熟悉CUDA/OpenCL编程优先;
5.有良好的学习能力、沟通能力和团队合作能力,工作态度端。

以担保或任何理由索取财物,扣押证照,均涉嫌违法,请提高警惕

立即申请