深圳虾皮信息科技有限公司招聘ai训练|推理框架工程师

深圳虾皮信息科技有限公司
招聘信息
AI训练/推理框架工程师
2026-08-06 14:48:04
职位描述
工作职责：
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 相关算子开发、调试经验，对 NCCL，cuDNN 等有一定了解；
5、熟悉 AI 模型训练、微调、评估算法和流程，具有相关项目经验。
职位类别:计算机软、硬件/互联网/IT
专业要求:工学,理学
单位简介
联系方式
公司地址