logologo
寻找工作
返回简章2025-12-17 更新

AI Infra 实习生 - 框架方向

北京
硕士及以上
计算机类·电子信息类
使用简历深度优化功能,快速提升简历质量
职位介绍
职位描述 1、 参与打造面向自动驾驶场景的算法研发框架,覆盖数据处理、模型训练、评测与部署等全流程。 2、 以高标准的工程规范参与日常研发工作,包括功能开发、文档撰写、CI/CD 流程建设,推动算法研发体系化、规范化。 3、 研究并解决大规模分布式训练中的性能瓶颈问题,涵盖数据加载、算子优化与通信加速等方向。 4、 沉淀通用算法模块(如多任务学习 Multi-Task、量化 QAT、模型蒸馏等),并推动其在自动驾驶业务中的落地与优化。 职位要求 1、 本科及以上学历,计算机、软件工程、人工智能等相关专业,能实习至少 6 个月优先。 2、 具备扎实的 Python / C++ 编程能力,代码规范良好,具备一定工程化经验。 3、 熟悉 PyTorch / TensorFlow 等主流深度学习框架,理解基本的模型训练与推理流程。 4、 对自动驾驶行业及 AI 基础设施方向有浓厚兴趣,具备快速学习和解决问题的能力。 加分项 1、 对 OpenMMLab、HuggingFace 等主流算法研发框架有深入理解。 2、 了解数据并行、模型并行、流水并行、混合并行等分布式训练策略。 3、 对 CUDA、NCCL、OpenMPI 等并行计算/通信框架有一定实践经验。 4、 对算法工程化与性能优化有独到见解,并有实际项目经验。 5、 有知名开源项目贡献经验。