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