NIO

校招-大模型训练框架开发工程师

NIO  •  Onsite  •  3 hours ago
Apply
AI can make mistakes so check important info. Chat history is never stored.

Job Description

校招-大模型训练框架开发工程师

北京

校招

正式

数字技术 - 算法

硕士及以上

2027届校园招聘-正式批

职位 ID:A47405

职位描述

参与智能辅助驾驶大模型训练链路的工程优化,主力支持以下一个或多个方向: 训练性能优化:分布式训练调优(显存/通信/IO),前后处理环节加速(解码/增强/加载),GPU/CPU异构计算优化 训练框架维护:内部训练框架迭代与模块化重构(数据加载/模型/优化器/日志解耦),支持增量学习与持续训练 数据链路优化:大规模数据读取链路设计(WebDataset/LMDB/TFRecord),预处理流程优化,解决分布式IO瓶颈 CI/CD与质量体系:自动化构建、测试与回归验证流程建设,训练效率与稳定性度量 系统问题攻关:内存/显存/日志/profiling专项优化,定位解决OOM、卡死、通信超时等系统级问题

职位要求

教育背景 硕士及以上学历,2027届应届生,计算机/软件/AI/自动化/电子等相关专业 必备能力 Linux/Git:熟练Linux开发环境,熟悉Git及CI/CD基本流程 C++:熟悉C++11/14/17标准,掌握内存模型、智能指针、STL、多线程编程 Python:熟悉多进程/多线程/协程,理解GIL与内存管理机制 PyTorch:熟悉PyTorch及分布式训练基本概念,有课程/项目实操经历 加分项 CUDA/GPU加速实践经历 了解LMDB/Arrow/WebDataset等数据格式 了解DDP/FSDP/DeepSpeed/Megatron等分布式策略 接触过Perf/Nsight/Triton/ ASan等性能工具 有训练框架阅读或小改动经历 有智能辅助驾驶/机器人/大模型训练相关项目或竞赛经历

投递

NIO

About NIO

NIO is a pioneer and a leading company in the premium smart electric vehicle market. Founded in November 2014, NIO’s mission is to shape a sustainable and brighter future together. NIO aims to build a community starting with smart electric vehicles to share joy and grow together with users.

NIO designs, develops, jointly manufactures and sells premium smart electric vehicles, driving innovations in next-generation technologies in autonomous driving, digital technologies, electric powertrains and batteries. NIO differentiates itself through its continuous technological breakthroughs and innovations, such as its industry-leading battery swapping technologies, Battery as a Service, or BaaS, as well as its proprietary autonomous driving technologies and Autonomous Driving as a Service, or ADaaS.

NIO’s models for sale include the all-new smart electric flagship SUV ES8, the smart electric flagship coupe SUV EC7, the smart electric mid-large SUV ES7, the smart electric flagship sedan ET7, the all-new smart electric all-round SUV ES6, the all-new smart electric coupe SUV EC6, the smart electric mid-sized sedan ET5, and the smart electric tourer ET5T.

Industry
Unknown
Company Size
5,001-10,000 employees
Headquarters
Jiading, CN
Year Founded
Unknown
Website
nio.com
Social Media