AI训练/推理框架工程师

20-40K/月 · 深圳南山 · 经验不限 · 本科

警示:以招聘为由收取培训费,提供培训贷款,缴纳体检、服装、押金等费用,都属违法行为;一经发现,请立即向110或12333举报!

职位描述

工作职责:
1、参与开发和建设公司的训练、微调、评估、推理工具链,支撑公司 AI 模型的分布式训练和推理应用;
2、参与公司内分布式训练和推理任务优化,包括但不限于框架优化、算子优化、通信优化和存储优化等;
3、与 AI 模型训练、推理业务团队以及平台等团队紧密协作,共同构建端到端的 AI 模型能力;
4、跟进和探索前沿的分布式训练、微调、评估和推理技术方案,推动相应工具的不断迭代。

岗位要求:
1、2026届应届毕业生,本科及以上学历;
2、掌握计算机专业基础知识,如计算机操作系统、计算机网络、计算机组成原理等;
3、掌握常见的数据结构和算法,编程能力良好,能够利用计算机程序解决实际的问题;
4、具有良好的团队合作精神,能够与跨部门的团队紧密合作,共同推动项目的成功;
5、具备较强的自我驱动力和学习能力,并乐于不断尝试、追求业务突破。

加分项:
1、熟悉常见的 CV、NLP 模型,如 ViT、GPT 等;
2、熟悉 AI 领域常见分布式训练技术的原理和实现,包括但不限于:数据并行、流水线并行和张量并行等,并具有相应的项目经验;
3、熟悉一种或多种深度学习框架,如 Pytorch、Tensorflow、PaddlePaddle、Deepspeed、Megatron 等,能够熟练使用并解决使用中遇到的问题;
4、熟悉 GPU 硬件结构和 CUDA 计算的原理,有 CUDA 相关…

工作地址

广 东深圳市南山区广东省深圳市南山区科技南一路28号达实大厦29楼Shopee

查看地图
登录后投递

请先登录后再投递简历