AI 系统软件工程师（高性能计算 / JAX 生态方向）
招聘年限：
关于我们与这个机会：
1、我们是一家走在技术前沿的芯片公司，致力于为人工智能时代打造最强大的计算基石。我们深知，卓越的硬件需要一个繁荣、开放、高效的软件生态来释放其全部潜能。因此，我们正在组建一支精英团队，开启一项激动人心的事业：以我们的自研芯片为核心，围绕 JAX 框架构建一个世界级的开源软件生态，并补全和完善从底层库到上层应用的 AI Infra 全链路；
2、这个职位并非传统的“螺丝钉”。我们不强行限定你的背景——无论你来自算法、框架、编译器、分布式系统还是基础设施领域，只要你对高性能计算（HPC） 抱有极致的热情，对打造有全球影响力的开源项目怀有梦想，这里就是你的舞台。在这里，你将有机会重新思考和定义 AI 时代的软件栈。我们鼓励探索，拥抱开源，并将在日常工作中积极探索和应用 AI Coding，让你始终站在技术浪潮之巅。
岗位职责:
1、生态构建与开源贡献： 深入参与开源社区，贡献核心代码，开发关键组件与库，使其在我们的硬件平台上发挥极致性能。你将是我们连接全球开发者的技术大使；
2、全链路性能优化： 作为一家芯片公司，我们追求极致性能。你需要与硬件、编译器团队紧密协作，打通软件与硬件的任督二脉，进行从模型算子到分布式训练的全链路性能分析与优化；
3、AI 基础设施建设： 参与设计和开发支撑大规模模型训练与推理的 AI 平台，包括但不限于资源调度、分布式训练框架、存储和网络方案等，并将其中的关键模块开源，回馈社区；
4、技术布道与社区互动： 撰写技术博客、分享设计文档、参与技术峰会，与全球的开发者交流，建立并扩大我们在 AI 开源社区的影响力；
5、前沿技术探索： 持续追踪 AI 系统与 HPC 领域的最新进展，并将其应用到我们的技术栈中，确保我们的生态始终保持领先。
任职要求:
1、对技术有无法抑制的热爱与好奇心，享受解决复杂技术挑战带来的成就感；
2、对高性能计算（HPC）充满兴趣，对压榨硬件性能、提升软件效率有极致的追求；
3、认同开源理念，有强烈的意愿和热情去构建一个开放、协作的技术生态；
4、专业背景与经验 (我们尤其欢迎具备以下一项或多项经验的你:
AI 框架层： 深入理解 JAX, PyTorch, TensorFlow 等框架的内部机制，有框架二次开发、算子优化或模型并行/分布式训练相关经验；
编译器技术层： 熟悉 MLIR, LLVM, TVM, Triton 等，有AI编译器开发、代码生成或底层性能优化经验；
高性能计算/库层： 熟悉 CUDA/ROCm/SYCL，精通并行计算模型，有高性能计算库（如 BLAS, NCCL, MPI）的开发或优化经验；
分布式系统/Infra层： 拥有大规模分布式系统或AI基础设施（如 Kubernetes, Ray, 高性能网络/存储）的设计与开发经验；
算法应用层： 作为算法工程师，但对底层系统和性能优化有浓厚兴趣并有相关实践经验。
如果你渴望在 AI 与 HPC 的交汇点上创造价值，并愿意与我们共同打造一个开放共赢的未来，请立即申请，我们期待与你的交流！