AI基础软件设计专家(算子)-杭州
沐曦股份-U · Software
岗位职责: 1. 负责主流 AI 计算框架(如Pytorch、Onnxruntime、Paddle、Tensorflow、MindSpore等)下算子的性能分析、优化策略制定与实施,提升算子在各类计算任务中的执行效率,降低资源消耗。 2. 负责主流LLM推理框架(如vLLM、sglang等)下算子的性能分析与优化。 3. 负责主流训练框架(如Megatron、DeepSpeed等)下算子的性能分析与优化。 4. 负责各种AI相关库的适配与算子性能优化,比如Open-CV、CV-cuda、MMCV、TritonServer等。…