【Plan A】Agent研究员-蚂蚁健康（实习）

岗位职责

部门介绍： 在蚂蚁，用 AI 守护生命：蚂蚁健康事业群智能工程引擎团队诚聘精英！每一个医疗决策的背后，都关乎生命质量。蚂蚁健康事业群正致力于通过 AI 技术实现医疗的真正平等化与普惠化。我们的智能工程引擎团队是实现这一愿景的基石。我们专注于打造业界领先的医疗 AI 基础设施，涵盖了从医疗领域大模型架构研发、训练推理性能极致优化到前沿智能体架构探索的完整链路。在这里，我们不设边界：你可以向下钻研 CUDA 级别的性能极限，向上探索最前沿的训练范式/模型架构。不要犹豫，与我们一起，用技术让普惠医疗触手可及！

职位描述：我们正在寻找能够定义下一代医疗 AI 智能体基础设施的核心成员。Agentic Infra 旨在构建高可用、可扩展且硬件感知的系统，支撑长程记忆架构、持续学习以及复杂的 Reasoning 工作流，让 AI 具备在各种医疗场景闭环执行任务的能力。

1. 记忆与状态架构研究：探索并实现类似 Engram 或长程上下文管理方案，解决 Agent 在复杂任务中的状态漂移与遗忘问题；
2. 持续学习基础设施：设计并优化大规模分布式持续学习训练框架，实现模型内部知识的快速更新；
3. 端到端系统协同设计：开发并优化高性能医疗智能体工作流，包括工具调用、多轮反思等；
4. 硬件敏感型内核开发：针对 Agent 频繁调用的底层算子（如 Sparse Attention），利用 TileLang 或 Triton 编写高效内核，提升吞吐并降低延迟。

任职要求

1. 计算机科学、人工智能、高性能计算或相关专业在读的硕士/博士；
2. 精通 C++/Python，具备 CUDA 或 TileLang/Triton 编程能力；
3. 熟悉持续学习/在线学习/强化学习原理；
4. 有在开源高性能训推框架贡献代码经验者优先；
5. 有在计算机顶会发表过关于模型架构优化、分布式系统或 Agent 算法论文者优先。

特色标签

AI医疗、AI基础设施、AI智能体、C/C++、GPU内核编译器、GPU加速、Python、Triton、Triton编译器、分布式训练、动态学习、医疗、医疗AI基础设施、医疗人工智能、医疗智能体、医疗领域大模型、发表算法相关优秀论文、在线学习、增量学习、复杂推理、多步推理、大模型、大模型算法、大语言模型、并行计算、强化学习、推理、显卡编程、智能体、智能体基础设施、智能医疗、机器学习、模型加速/性能优化、注意力稀疏化、深度学习、稀疏注意力、稀疏注意力机制、算法工程化经验、自然语言处理算法、高性能GPU算子开发工具