职位详情
智算开发工程师(AI方向)
2-3万·15薪
上海张江数学研究院
上海
不限
硕士
11-03
工作地址

张江国创中心一期-C4号楼

职位描述
岗位职责:

1、负责大模型训练与推理系统的研发与优化,提升模型在GPU/国产AI芯片上的运行效率;

2、深入理解主流AI框架(如PyTorch、MindSpore、TensorFlow等),参与大模型训练/推理框架的定制开发与性能调优;

3、研究并实现大模型推理加速技术,包括模型量化、剪枝、蒸馏、算子融合、图优化等;

4、参与分布式训练与推理系统的设计与实现,支持超大规模模型的多机多卡部署;

5、跟踪前沿AI系统优化技术,结合业务需求进行技术预研与落地验证;

6、与算法团队协作,推动大模型在气象、医疗、制造等垂直领域的应用落地。


任职要求:

1、计算机、人工智能、软件工程、数学等相关专业硕士及以上学历;

2、熟练掌握Python/C++,具备良好的编程能力和系统调试经验;

4、熟悉主流深度学习框架(如PyTorch、TensorFlow)及其底层实现机制;

5、熟悉大模型训练/推理加速技术,具备TensorRT、FasterTransformer、vLLM、DeepSpeed、Megatron等框架使用经验者优先,熟悉当前主流大模型的架构特点;

6、熟悉GPU编程(CUDA)或国产AI芯片(如昇腾、沐曦、海光、寒武纪)开发者优先;

7、具备良好的团队协作能力和沟通能力,有较强的学习能力和技术热情。

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

立即申请