NVIDIA 致力于开发加速计算和人工智能技术，广泛应用于游戏、数据中心、云计算、医疗保健、制造、汽车和机器人等领域。其业务涵盖图形处理器、人工智能平台、仿真技术及面向行业的解决方案，包括支持协作式 3D 工作流程的 NVIDIA Omniverse、用于自动驾驶汽车开发的 NVIDIA DRIVE，以及面向医疗应用的 NVIDIA Clara。公司汇聚了众多技术团队，专注于构建支持先进计算、仿真和人工智能驱动分析的基础设施与软件。
深度学习编译器高级工程师——CUDA Tile 与 LPU
为 NVIDIA 的 AI 与 GPU 计算平台开发 CUDA Tile 转换和 MLIR 降低阶段。 提升适用于 GPU 与 LPU 架构的基于 Tile 的内核性能。
描述
- 支持 CUDA Tile——NVIDIA 面向 LPU 开发的基于 Tile 的编程模型
- 设计并实现编译器转换
- 创建 MLIR 方言和降低阶段
- 提升适用于多代 NVIDIA GPU 与 LPU 架构的基于 Tile 的内核性能
- 定义公共 API
- 开发编译器与优化方法
- 负责性能调优及更广泛的软件工程任务
- 与专家团队合作，开展服务于深度学习社区的工作
任职要求
- 拥有计算机科学、计算机工程或相关专业的学士、硕士或博士学位，或具备同等经验
- 至少三年编译器优化、性能分析和中间表示设计方面的相关经验或研究经历
- 能够自主工作、制定项目目标和范围，并负责个人开发工作
- 具备高级 C 和 C++ 编程及软件设计能力，包括调试、性能分析和测试规划
- 具备良好的沟通能力，能够在快速推进、以产品为导向的团队中有效协作
- 熟悉 CPU 和/或 GPU 架构者优先
- 具备 CUDA 或 OpenCL 编程经验者优先
- 具备 MLIR、LLVM、XLA、TVM、深度学习模型和深度学习算法经验者优先
福利待遇
- 股权薪酬
- 全面的福利待遇