校招-大模型训练框架开发工程师
北京
校招
正式
数字技术 - 算法
硕士及以上
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 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.