logo of amap

高德地图高德-AI Infra 训练与推理引擎优化专家-北京

社招全职3年以上地点:北京状态:招聘♡ 收藏

工作描述


任职要求
1、本科及以上学历,计算机科学、软件工程或相关专业,具备系统软件、高性能计算或 AI 基础设施方向的研发经验。
2、精通 C++/Python,具备扎实的多线程、内存管理与性能剖析能力,熟悉 CUDA 编程模型与 GPU 体系结构,能独立完成算子级性能调优。
3、深入理解主流训练框架(Megatron-LM、DeepSpeed、PyTorch FSDP)或推理引擎(vLLM、SGLang、TensorRT-LLM)的实现原理,具备二次开发或自研落地经验。
4、具备千卡及以上规模分布式训练集群的实战经验,熟悉 NCCL、RDMA 等高性能通信机制,能定位并解决通信瓶颈、显存溢出与训练不稳定问题。
5、熟悉模型量化、稀疏化与编译优化(TVM、MLIR、Trit…
登录查看完整工作描述
微信扫码,1秒登录

包括英文材料
学历+
C+++
Python+
多线程+
CUDA+
性能调优+
Megatron+
FSDP+
推理引擎+
大模型+
还有更多 •••
相关职位

logo of xpeng
校招

基本技能要求: 1、计算机科学/计算机工程/电子工程或相关专业硕士学位,或具备同等的行业经验。 2、精通 PyTorch。 3、熟悉模型推理框架(如 vLLM、SGLang)。 4、深入理解 Tran

更新于 2026-09-23上海|广州|北京
logo of tencent
社招1年以上搜一搜技术

1.具备较强的动手能力;熟悉 Python ,具备扎实的系统编程功底和优秀的复杂系统 Debug 能力; 2.深入理解大模型分布式训练原理,具备 Megatron-LM、DeepSpeed 或 Py

更新于 2026-10-13广州
logo of weride
校招

岗位简介 加入我们,你将参与国内头部自动驾驶公司万卡级GPU训练集群的构建与优化,直接支撑端到端自动驾驶大模型的训练迭代。在这里,你会深入PyTorch编译器、CUDA Kernel、NCCL通信库的

更新于 2026-07-07深圳|广州|北京
logo of antgroup
校招2027届蚂蚁星

1.熟悉 Python、C++ 中至少一种语言,具备良好的系统设计和工程实现能力; 2. 熟悉 PyTorch 训练机制,对分布式训练、并行策略、显存优化或高性能计算有实践经验; 3. 对 Megat

更新于 2026-07-28北京|上海|杭州