独角兽

软件库性能优化工程师/架构师

壁仞科技

北京 / 上海 / 杭州 / 其他人工智能社会招聘

岗位描述

岗位描述:负责AI模型算子的开发与优化;负责推理或训练模型的端到端调优;与软硬件架构师共同开展软件流程及架构探索;推动并解决跨团队性能问题。 岗位要求:计算机、电子信息类专业本科及以上学历;具备CUDA、OpenCL等高性能并行编程能力;熟悉AI模型算子开发和优化;熟悉AI模型推理或训练场景的端到端优化;熟悉深度学习算法和主流AI模型,如RN50、BERT、GPT、LLaMA等;对计算机体系结构有深刻理解;精通Nsight Compute等性能分析工具;精通PyTorch、PaddlePaddle等至少一种深度学习框架;熟悉常用benchmark工具,如Sysbench、iometer、netperf;熟悉常用性能分析工具,如SystemTap、perf、vtune、gperf、oprofile、火焰图。