更新于 11月3日

智算开发工程师(AI方向)

2-3万·15薪
  • 上海浦东新区
  • 经验不限
  • 硕士
  • 全职
  • 招1人

职位描述

人工智能计算机软件
岗位职责:
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、具备良好的团队协作能力和沟通能力,有较强的学习能力和技术热情。

工作地点

上海浦东新区张江国创中心一期-C4号楼

职位发布者

吕先生/HR

立即沟通
公司Logo上海张江数学研究院
上海张江数学研究院(以下简称“数学院”)是事业单位性质的新型科研机构。数学院聚焦“数学+”,开展相关领域人才培养、科学研究、成果转化、产业化等。支撑上海国际科技创新中心、张江综合性国家科学中心和张江国家实验室建设,助推浦东新区高水平改革开放,打造社会主义现代化引领区建设,努力建设成为数学服务国家重大战略任务的国家队。
公司主页