理想汽车
大模型训练推理加速实习生
RESPONSIBILITIES
岗位职责
负责大模型训推框架优化加速/自动驾驶模型训练系统优化;
大模型优化方向预研,如可训练稀疏注意力、结构化稀疏等;
强化学习训练框架开发。
REQUIREMENTS
任职要求
熟悉C++/python,有PyTorch使用经验;
了解CV/LLM/RL主流模型,熟悉常用的训练/推理优化手段;
有大规模并行计算工程经验,CUDA性能优化经验优先;
熟悉AI编译器,有LLVM、TVM、MLIR、XLA 等编译开发经验者优先;
熟悉大规模并行计算集群、分布式存储基础架构、高性能Collective 通信原语等优先。
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。