CUDA算子开发工程师(J10697)

1.负责CUDA算子的开发适配和性能调优,支持复杂模型在服务器端的训练与推理 2.主导解决定制化算子在训练框架、AI编译工具和部署应用中的问问题,确保算子在GPU、CPU及端侧异构计算平台上的高效计算和稳定兼容 3.跟踪算子领域新技术,根据AI模型发展趋势和硬件特性持续丰富算子库、提升算子性能 4.负责高性能算子的开发与优化,包括CPU、DSP、RVV等异构硬件,优化算子的性能,确保在不同硬件平台上的高效计算,包括但不限于FP32、FP16、FP8、INT16、INT8、INT4等数据类型 【任职要求】 基本要求 1.具备丰富的GPU程序开发经验,熟悉CUDA、OpenCL,熟悉GPU加速库,包括并不限于cuBLAS/cuDNN/cutlass 2.熟悉各类深度学习网络和算子底层实现细节,有模型训练、推理调试调优经验 3.具备良好的工程实现能力,熟悉Python、PyTorch与C/C+…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。

本站当前在招 22528 个上海岗位（全城各职能合计），其中 2290 个公开标注了薪资。按这些标注统计：月薪（税前）主流区间 8.4K ~ 20K（25~75 分位），中位约 12K，最低 0.8K、最高 120K。