岗位职责
- 在AI 芯片上适配SOTA开源框架和SOTA模型,分析解决适配过程中发现的功能、性能与精度问题,为AI芯片客户提供问题支持和解决方案;
- 对SOTA模型和框架做原理分析,竞对分析、构建差异化产品竞争力,贡献开源社区提升产品影响力;
- 构建SOTA模型和框架端到端粒度、算子粒度、Kernel粒度的仿真框架,在芯片设计阶段提前发现精度、性能问题,保障芯片设计成功。
职位要求
- 硕士及以上,计算机科学相关专业, AI领域相关的研究/实习经验;
- 熟练掌握Python、C/C++编程语言,掌握AI Vibe Coding技术,良好的逻辑思维与问题分析能力;
- 理解SOTA大语言模型、多模态模型、VLA模型的架构与原理;
- 理解主流深度学习框架Pytorch、Tensorflow,分布式训练框架Megatron、DeepSpeed,推理框架VLLM、SGlang等,增强学习框架 Slime 、roll 等;
- 理解CUDA、Triton、Tilelang 并行编程技术;
- 责任心强,具备跨团队协作意识和沟通交流能力。