蚂蚁金服蚂蚁集团-AI基建工程师（引擎方向）-北京/杭州
社招全职2年以上技术类-开发地点：北京 | 杭州状态：招聘♡ 收藏
工作描述
任职要求 1. 具备分布式系统设计与实现经验，熟悉Ray分布式框架，熟悉SPMD编程模型和集合通信原理，具备通信优化（NCCL/RDMA）和并行策略（DP/TP/PP/EP等）实现与调优能力； 2. 精通PyTorch框架，熟悉transformer网络结构和原理，掌握Transformers、flash-attention等主流库的使用和实现细节； 3. 熟悉计算图构建、算子融合、显存布局优化、CUDA高性能算子…
登录查看完整工作描述
微信扫码，1秒登录
包括英文材料
分布式系统+
https://www.distributedsystemscourse.com/
The home page of a free online class in distributed systems.
https://www.youtube.com/watch?v=7VbL89mKK3M&list=PLOE1GTZ5ouRPbpTnrZ3Wqjamfwn_Q5Y9A
Ray+
https://github.com/ray-project/ray
Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
https://www.youtube.com/watch?v=FhXfEXUUQp0
In this video, I'll teach you everything you need to know about Apache Ray!
https://www.youtube.com/watch?v=fMiAyj2kgac
Using powerful machine learning algorithms is easy using Ray.io and Python.
https://www.youtube.com/watch?v=q_aTbb7XeL4
Parallel and Distributed computing sounds scary until you try this fantastic Python library.
NCCL+
https://developer.nvidia.com/nccl
The NVIDIA Collective Communication Library (NCCL) implements multi-GPU and multi-node communication primitives optimized for NVIDIA GPUs and networking.
PyTorch+
https://datawhalechina.github.io/thorough-pytorch/
PyTorch是利用深度学习进行数据科学研究的重要工具，在灵活性、可读性和性能上都具备相当的优势，近年来已成为学术界实现深度学习算法最常用的框架。
https://www.youtube.com/watch?v=V_xro1bcAuA
Learn PyTorch for deep learning in this comprehensive course for beginners. PyTorch is a machine learning framework written in Python.
Transformer+
https://huggingface.co/learn/llm-course/en/chapter1/4
Breaking down how Large Language Models work, visualizing how data flows through.
https://poloclub.github.io/transformer-explainer/
An interactive visualization tool showing you how transformer models work in large language models (LLM) like GPT.
https://www.youtube.com/watch?v=wjZofJX0v4M
Breaking down how Large Language Models work, visualizing how data flows through.
CUDA+
https://developer.nvidia.com/blog/even-easier-introduction-cuda/
This post is a super simple introduction to CUDA, the popular parallel computing platform and programming model from NVIDIA.
https://www.youtube.com/watch?v=86FAWCzIe_4
Lean how to program with Nvidia CUDA and leverage GPUs for high-performance computing and deep learning.
Megatron+
https://www.youtube.com/watch?v=hc0u4avAkuM
还有更多 •••
相关职位
社招3年以上后端开发
1、统招本科及以上学历，计算机相关专业，3年以上软件开发经验，有架构设计经验优先； 2、Java基础扎实，对IO、多线程、JVM等有深入了解，熟练使用常见编程框架，并对其底层原理有深刻理解； 3、具备
更新于 2026-06-10北京|上海
社招3年以上
1、计算机科学、人工智能、软件工程或相关专业本科及以上学历； 2、熟练掌握 Java/Python 等编程语言，具备扎实的数据结构、数据库、操作系统、计算机网络等基础理论知识，以及较强的学习能力； 3
更新于 2026-06-29杭州
社招3-5年
1、本科及以上学历（重视潜力，不严格限制专业），3-5 年开发经验，基础扎实。 2、扎实的前端功底：精通 HTML / CSS / JavaScript / TypeScript，熟悉 React 或
更新于 2026-08-27北京
社招2年以上
1、计算机科学、人工智能、软件工程或相关专业本科及以上学历； 2、熟练掌握 Java/Python 等编程语言，具备扎实的数据结构、数据库、操作系统、计算机网络等基础理论知识，以及较强的学习能力； 3
更新于 2026-01-29杭州