顶尖实习-大模型高性能算子研发工程师-AI基建当面告知
小米科技有限责任公司
2027-02-08 23:59:59
博士27届28届实习硕士
岗位详情
地区：
北京
专业要求：
软件工程
岗位职责：
- 基于 CUDA / CUTLASS 以及 Triton / TileLang 等算子开发工具，针对不同模型结构与硬件架构进行算子设计、实现、融合（kernel fusion）与调优，提升计算访存效率与硬件利用率（MFU）
- 使用 nsys / ncu 等性能分析工具定位计算、访存、通信瓶颈，开展 roofline 分析与 kernel 级优化，持续提升推理吞吐、降低时延。
- 跟踪业界前沿（新模型结构、新硬件特性、开源算子库等），参与算子库开发和建设、性能基准测试与正确性验证
招聘要求：
- 硕士及以上学历，计算机、软件工程、数学等相关专业，具备扎实计算机理论基础和丰富编程经验，在深度学习领域有深入研究与实践。
- 精通 Python，熟练掌握至少一种深度学习框架（如 PyTorch），理解深度学习和transformer系列算法，有丰富模型开发与调优经验，能独立搭建复杂模型并优化性能。
- 熟悉主流的并行编程以及性能优化技术，熟悉CUDA和TensorCore编程，有相关高性能优化经验(nsys/ncu)优先；
- 熟悉flashattention，flashinfer， deepgemm等算子库， 有使用triton/tilelang等做高性能算子开发经验优先
- 有良好团队协作与沟通能力，能与不同背景成员有效沟通协作，责任心强、敬业热情，能承受工作压力，保证项目按时高质量交付。
公司介绍
小米科技有限责任公司
小米科技有限责任公司，成立于2010-03-03，位于北京，法定代表人：雷军，登记状态：存续，注册资本185000万人民币，公司类型：民营/私企
"小米科技有限责任公司成立于2010年3月3日，是专注于智能硬件和电子产品研发的全球化移动互联网企业，同时也是一家专注于智能手机、智能电动汽车、互联网电视及智能家居生态链建设的创新型科技企业。小米公司创造了用互联网模式开发手机操作系统、发烧友参与开发改进的模式。“为发烧而生”是小米的产品概念。“让每个人都能享受科技的乐趣”是小米公司的愿景。小米公司应用了互联网开发模式开发产品，用极客精神做产品，用互联网模式干掉中间环节，致力让全球每个人，都能享用来自中国的优质科技产品。小米已经建成了全球最大消费类IoT物联网平台，连接超过4.78亿台智能设备，进入全球100多个国家和地区。MIUI全球月活跃用户达到5.5亿。小米系投资的公司超500家，覆盖智能硬件、生活消费用品、教育、游戏、社交网络、文化娱乐、医疗健康、汽车交通、金融等领域。"