蚂蚁金服蚂蚁集团-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北京