AI 芯片加速算法与软件工程师
- NIO
- Shanghai, Shenzhen +2 more, China
- CNY 400,000 – CNY 700,000
上海、深圳、杭州、合肥社招全职数字技术 - 芯片研发本科及以上1-3 年
职位描述
- 开展芯片与算法软硬件协同优化，针对Dense/MOE模型场景，优化推理精度、时延、吞吐、显存占用等核心指标，参与芯片架构需求定义；
- 结合自研异构算力芯片架构特性，参与 AI Infra软件栈研发，完成高性能算子开发、计算图优化、显存调度、算子融合等核心工作，最大化释放芯片硬件算力，支撑大模型、具身智能等业务落地；
- 负责深度学习训推引擎适配、主流模型移植与性能精度调优，搭建自动化性能测试、精度验证体系，定位并解决模型部署、版本迭代中的性能退化、精度异常等问题；
- 跟踪 AI 编译器、异构计算、量化稀疏、高性能算子加速等前沿技术，开展技术预研与原型验证，沉淀技术方案、专利成果，推动 AI 加速软件栈持续迭代与产品化落地。
职位要求
- 学历专业：本科、硕士及以上学历；
- 工作经验：1-5 年及以上 AI 芯片加速、算子优化、推理引擎、异构计算软件开发相关工作经验，有算力芯片软件栈落地经验者优先；
- 技术能力：熟悉 C/C++、Linux 系统编程，熟练使用 Python，具备扎实的底层软件开发、调试与性能调优能力。熟练掌握AI coding高效开发经验者优先；
- 熟悉计算机体系结构、异构计算原理与深度学习基础，了解 PyTorch、ONNX 等主流框架，具备算子开发、计算图优化、显存调度优化实战能力；有 CUDA/NPU 算子库开发、模型量化、编译器优化经验者优先；
- 具备大模型、具身智能场景模型部署与加速优化经验，熟悉推理训练框架二次开发者优先，有国产算力芯片软件栈自研项目经验者优先。
投递
Skills
- C/C++
- CUDA
- PyTorch
- ONNX
- AI Compiler
- Performance Optimization
- Heterogeneous computing
