返回岗位列表

小米

RISC-V计算架构与并行计算架构底层算子库开发工程师

地点:北京 薪资:薪资未公开 日期:2026-01-29

岗位摘要

基于RISC-V计算架构、并行计算架构,负责设计、开发、优化底层算子库,提升底层芯片执行性能;研究和探索LLM大模型、扩散模型等前沿深度学习算法在先进计算架构芯片上的优化算法,通过算法优化和硬件优化结合,实现更高的算法性能

岗位职责

  • 基于RISC-V计算架构、并行计算架构,负责设计、开发、优化底层算子库,提升底层芯片执行性能
  • 研究和探索LLM大模型、扩散模型等前沿深度学习算法在先进计算架构芯片上的优化算法,通过算法优化和硬件优化结合,实现更高的算法性能

任职要求

  • 具备较强的工程能力,熟悉C/C++/Python,熟悉tensorflow/pytorch/onnx主流AI框架
  • 深入了解常见的LLM大模型,例如GPT、BERT、LLAMA等,理解内部模型结构原理和优化方法,具备相应的实践经验
  • 具备在CUDA/cuDNN/TensorRT/nccl等高性能库上实现和优化高性能算子的开发经验
  • 熟悉并行/异构计算体系结构,熟悉CPU/GPU/NPU/TPU等主流XPU架构和编程模型优先
  • 在AI相关顶刊顶会发表论文者更优
  • 良好的团队沟通能力、技术攻关能力,能独立思考和解决问题,责任心强,工程师素养佳