2027校招-算子开发工程师(上海/北京/成都) 上海、北京、成都 校招 正式 职位描述 岗位职责- 负责公司自研 AI 加速芯片上的 算子库设计与实现,覆盖训练与推理核心算子(如 GEMM/Conv/Attention/LayerNorm/Softmax/Reduce/Scatter-Gather 等)。- 基于芯片计算单元、片上存储(SRAM/Cache)、HBM 带宽、NoC 拓扑、DMA/异步拷贝能力,进行 算子性能建模与瓶颈分析(roofline、访存/计算比、流水并行),制定优化策略。- 设计并落地 kernel 调度与 tiling 策略:多级分块、数据布局(layout/packing)、向量化/张量化、双缓冲/多缓冲、算子融合(fusion)等。- 建设算子库工程化体系:算子接口规范、版本管理、性能回归、单测/对齐验证、精度与数值稳定性、跨平台兼容- 参与端到端模型性能优化:对接 PyTorch/TensorFlow/ONNX Runtime 等前端(或公司自研框架),完成算子覆盖、性能调优和线上问题定位(profiling/trace)。 职位要求 (优秀应届可放宽)。- 计算机/电子/自动化/数学等相关专业;- 熟悉深度学习常见算子与数值实现(GEMM/Conv/Attention/Norm/Reduce),理解训练/推理的差异(激活、梯度、混合精度、量化)。- 扎实的 C/C++ 能力,良好的工程习惯;能熟练使用性能分析工具(profiling、trace、perf、火焰图或自研工具)。- 熟悉至少一种加速器编程模型或并行体系(CUDA/OpenCL/ROCm/Metal/TVM/Triton/oneDNN/CUTLASS 等)并能迁移到自研 ISA/Runtime。- 具备系统级性能意识:理解存储层次、带宽/延迟、cache/预取、NUMA、DMA、并行同步等,对算子优化有方法论。加分项-
We are now looking for a Deep Learning Performance Software Engineer! We are expanding our research and development for deep learning. We seek excellent Software Engineers and Senior Software Engineers to join our team. We specialize
新业务部-AI编译器开发工程师/专家 北京、成都 社招 全职 职位 ID:A196077 职位描述 1.负责AI芯片编译器的方案及开发实现(侧重点为高能效比与加速器的高利用率);2.负责编译器及相应工具链的开发和维护;3.结合AI算法,架构和产品需求,完成AI编译器的方案及软件开发;4.模型的性能优化,研发性能和功耗分析及优化工作5.结合业界编译器的发展动态,规划自研编译器的演进 职位要求 1.计算机,通信,微电子工程或相关专业硕士以上学历,本方向3年及以上工作经验;2.精通处理器的指令集,图编译,编译前端,中端优化技术,后端code生成,任务排序优化;3.熟悉TVM或者MLIR一种编译框架,熟悉网络的编译优化策略;4.精通C++编程,有GPGPU/NPU编译经验者优先;5.有成功IP落地的经验者优先 投递...
深度学习工具链研发工程师 北京、武汉 社招 全职 职位 ID:A127152 职位描述 1. 负责开发自动驾驶深度学习工具链2. 负责自动驾驶算法模型优化与部署工作3. 负责自动驾驶深度学习模型量化/压缩算法的研发 职位要求 1. 扎实的编程基础(Python/C++)与良好的工程习惯2. 熟悉常用深度学习框架(如PyTorch, TensorFlow等)3. 有深度学习模型训练、量化、裁剪、NAS、HPO、部署等任一经验者优先4. 熟悉编译优化技术(TVM/XLA/MLIR/Halide等)者优先5. 熟悉计算机视觉、自动驾驶感知算法者优先 投递...