[北京]江原芯科技（上海）有限公司

职位：AI芯片大模型算法实习生
发布时间：2025-07-16
工作地点：北京
信息来源：前程无忧(51JOB)
职位类型：全职
职位描述
专业1:计算机科学与技术
专业2:计算机技术
福利:毕业后转正
职能类别:算法工程师
岗位职责：
1. 参与AI领域的软件工程化和产品开发,协同学术界/工业界专家资源，探索具备高度适应性的高能效AI算法、架构与平台
2. 针对数据领域, 机器学习和深度学习等技术领域要有一定的技术深度,能够针对应用场景问题进行方案设计并在他人的辅助下或者独立解决
3. 参与大模型在自研AI芯片上的推理部署全流程，包括模型移植、性能调优和芯片指令集及集群拓扑架构的适配
4. 参与开发面向芯片架构和系统集群拓扑架构架构的模型优化方案（算子融合/量化/稀疏化/多机多卡并行方案/动态计算等），突破芯片算力瓶颈
5. 参与构建芯片感知的推理框架优化方案，包括内存管理、TP/DP/SP/EP等并行方案，面向KV Cache的调度优化，以及计算与通信融合的异构计算加速；
6. 参与设计大模型推理的能效比评估体系，赋能芯片-模型协同设计优化；
跟踪LLM前沿技术（MoE/多模态/长上下文等），推动芯片架构创新

任职要求：
1. 计算机相关专业，具备一个及以上大型实际软件项目经验，独立承担过关键子模块的开发工作；
2. 熟练运用至少一门编程语言（C++ /Java /Python/Rust等）；
3. 熟悉Tensorflow/scikit-learn/xgboost/Caffe/MindSpore/pytorch等主流深度学习框架的使用，具备AI算法开发及应用经验者优先
4. 精通Transformer/GPT/LLaMA/QWEN/Deepseek V3等大模型架构
5. 拥有cuda 芯片算子开发& 低比特量化能力的优先
6. 熟练掌握LLM推理框架（vLLM/SGLang/TensorRT-LLM/FastTransformer）经验的优先考虑
7. 硕士在读，26届优先 ,其他特别有潜力的也可以考虑。
8. 人比较聪明,机灵, 不会没问题,一定要态度好,肯学,喜欢沟通
9. 工作时间：期望每周5天全勤,保底实习6个月以上。长期实习,后续表现优秀可以走校招进入公司
10. .实习培养方向: AI算法工程实习 架构师方向" 公司简要介绍：
公司名称:江原芯科技（上海）有限公司
公司类型:民营公司
公司介绍:江原科技成立于2022年11月，致力于实现我国人工智能芯片供应链自主可控，成为国内领先的“全国产化生产、制造和全内循环”芯片企业，实现中国芯，中国造。公司核心成员来自AMD、英伟达等国内外龙头芯片企业，截止目前，公司已打造了一支百人规模的团队，其中研发人员占比90%以上。公司在北京、上海、杭州分别设有研发中心。致力成为异构计算的领军企业，通过芯片、软件和系统一体化设计，打造领先的异构融合计算平台与技术生态。