岗位描述(节选)
面向大模型训练/推理场景,开发和优化高性能 GPU 算子(Attention、GEMM、LayerNorm、MoE、Element-wise 等),逼近硬件理论峰值
针对新模型结构(长 context、MoE、线性 Attention 等)设计定制化 kernel,解决原生框架无法覆盖的性能瓶颈
深度优化访存效率、计算-访存重叠、算子融合(fusion),降低 kernel launch 开销与显存占用
参与推理引擎核心链路优化:KV Cache 管理、量化算子(INT8/FP8/INT4)、投机采样、continuous batching 等
与训练框架、模型算法团队紧密协作,将算法需求转化为高效的系统实现
探索新硬件(新一代 GPU / 国产算力)上的算子适配与性能调优
【任职要求】
基础要求
扎实的 CUDA 编程能力,深入理解 GPU 体系结构(SM、warp、shared me……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。
本站当前在招 21050 个北京岗位(全城各职能合计),其中 1782 个公开标注了薪资。按这些标注统计:月薪(税前)主流区间 7K ~ 20K(25~75 分位),中位约 12K,最低 1K、最高 300K。
在招岗位较多的企业:字节跳动 1498 个美团 1480 个阿里巴巴集团 1180 个快手 952 个小米集团 861 个
统计口径:求职者营地实时在招岗位的公开薪资标注,未标注者不计入;区间为 25~75 分位,非薪酬承诺。