AI infra应该学什么?

AI Infra 是未来 5–10 年非常确定的岗位方向之一,而且相比纯算法岗,门槛更工程化、需求更大、年龄更友好

AI Infra 技能栈

你可以把技能栈理解为 4 层:

第1层:基础工程能力(必须)

这是门槛:

  • Linux
  • Python / Go
  • 计算机网络
  • 操作系统
  • Docker
  • Kubernetes
  • 数据库(MySQL / Redis)
  • 分布式系统基础
  • API开发(FastAPI / Flask)
  • 并发 / 异步 / 队列(Kafka / RabbitMQ)

这一层本质是:后端工程师能力

第2层:AI工程能力

这一层开始进入 AI Infra:

  • PyTorch 基础
  • 模型推理(inference)
  • 模型部署(TorchServe / Triton)
  • ONNX / TensorRT
  • vLLM / TGI / LMDeploy
  • Embedding / 向量数据库MilvusFAISSWeaviatepgvector
  • RAG(检索增强生成)
  • Prompt 工程
  • Agent 框架LangChainLlamaIndexAutoGenCrewAI

第3层:系统能力(高薪区)

这一层是专家岗做的事:

  • GPU 调度
  • 分布式训练(DeepSpeed / FSDP)
  • 推理优化
  • KV Cache
  • Batch 推理
  • Token 优化
  • 成本优化
  • 高并发系统设计
  • 模型服务化(Model Serving)
  • 多 Agent 系统架构
  • Memory 系统
  • Tool 调用系统

第4层:AI 平台能力(更偏架构)

例如:

  • AI 平台
  • 数据平台
  • 训练平台
  • Agent 平台
  • Prompt 平台
  • Evaluation 平台
  • Feature Store
  • 实验平台

这一层已经是 架构师 / Infra负责人 级别。

对应岗位名称

你以后找工作不要只搜“AI工程师”,要搜这些:

常见岗位名:

  • AI Infra 工程师
  • 机器学习平台工程师(ML Platform)
  • MLOps 工程师
  • LLM Infra 工程师
  • 推理工程师(Inference Engineer)
  • 模型部署工程师
  • AI 后端工程师
  • AI 平台工程师
  • GPU / 分布式训练工程师
  • 向量检索工程师
  • RAG 工程师
  • Agent 平台工程师

公司 JD 关键词:

  • RAG
  • Agent
  • LLM
  • vLLM
  • Triton
  • TensorRT
  • 向量数据库
  • Kubernetes
  • 推理优化
  • 高并发
  • 分布式
  • GPU

看到这些词,基本就是 AI Infra 岗。

入行路径

路线1(最推荐):后端 → AI Infra

路径:后端工程师 → AI后端 → RAG工程师 → AI Infra工程师

先学:

  • Python
  • FastAPI
  • MySQL / Redis
  • Docker
  • Linux

然后做项目:

  • 做一个 RAG 问答系统
  • 做一个 Agent 调用工具系统
  • 部署一个开源模型
  • 做一个向量检索系统
  • 做一个多 Agent workflow

你只要能做出:

  • 文档问答系统(RAG)
  • Agent 自动调用工具
  • 模型部署 API
  • 高并发请求处理

就已经是 AI Infra 初级工程师水平 了。

建议做的项目方向

如果你真的想入行,建议做这 4 个项目写在简历上:

  1. 本地部署大模型 + API 服务vLLM / Ollama / LMDeployFastAPI 提供接口
  2. RAG 系统文档 → 切分 → embedding → 向量数据库 → 检索 → LLM生成
  3. Agent 系统LLM + Tool 调用例如:自动查天气 / 查数据库 / 发邮件 / 调接口
  4. 高并发测试压测缓存队列限流

你可以用这个判断自己是否进入 AI Infra:

如果你每天在做的是:

  • 调 Prompt → 偏应用
  • 训练模型 → 偏算法
  • 做 RAG → AI工程
  • 做 Agent → AI工程
  • 部署模型 → AI Infra
  • 优化推理速度 → AI Infra
  • GPU 调度 → AI Infra
  • 成本优化 → AI Infra
  • 高并发系统 → AI Infra

#从事AI岗需要掌握哪些技术栈?#
全部评论

相关推荐

肖先生~:这个题目简单的让人感觉到不可思议
点赞 评论 收藏
分享
今天 11:20
东南大学 C++
暑期面了腾子五次后台开发岗,不出意外是全都挂了,主要原因我总结下来大概是:1. 首先当然是自己学的背的还是不够多不够深,没有某一场面试能跟说答的上八九不离十,自己还是有很多知识盲区,感觉还是要先沉淀一下。2. 对于后台开发来说大多数是以golang和java为主要开发语言,虽然也面到过cpp但也是属于音视频领域了。自己做的cpp客户端实习和偏向于AI应用的项目整体和传统后端偏差可能比较大(仔细想了想简历匹配度上确实不太好,之前把时间放在学ai上就没有重新弄传统后端框架的项目),这我感觉也是很大的问题3. 环境问题,感觉暑期前期时间点的横向竞争比较大(全是大佬争offer),对比其他面试者如果有后端实习和对应的后端项目确实难以竞争。现在打算做个抉择(也希望佬给点建议):1. 暂时先将投递岗位转到移动客户端,虽然知道客户端前景没有后端好,但鉴于现在的大环境,转到后端也不能说无忧无虑(没有拉踩,之前相较于后端客户端就是💩),还是认为要向AI赋能方向上靠。2. 在这段时间将原本的项目重构出golang后端框架,加入主流技术栈,背八股,然后继续投后端后台岗位。自己是想同步进行换岗位面试和重构项目(利用ai整体速度还是很快的,主要强化自己的理解和八股会花很多时间),等雏形出来了就换简历上去(当然如果期间oc了可能也就考虑直接接了,有鹅选鹅,不想想那么多了,我做这行也就是想赚个年轻钱,能干几年十几年吧),还要准备学AI应用的进阶知识,还是有很多知识盲区(面试中有些问题会问的更深)自己的回顾和反思大概是这样,发这里记录和分享一下。
点赞 评论 收藏
分享
评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务