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岗需要掌握哪些技术栈?#
全部评论

相关推荐

昨天 16:49
门头沟学院 Java
点赞 评论 收藏
分享
3.27日更新:秒挂自我介绍Serverless 平台项目周期、是否有用户、是否有反馈项目的背景项目相比 Linux 还缺什么如果拿去实际用,你觉得它和 Linux 相比最核心缺什么,举一个场景你觉得问题出在哪、和 Linux 比有什么不同、有没有解决思路Linux 的机制已经足够完美了吗是否可以理解为 Linux 因历史兼容导致结构变重,而你们有机会做得更轻Linux 在处理并发/锁问题上是怎么做的有没有做过 Linux 和当前系统在调度/进程管理上的开销对比测试运行过程中的 limit 是如何控制的项目到现在为止是否使用了 Rust 的 unsafe,如果用了如何处理内存隐患cgroup 在 Linux / 操作系统层面是怎么生效的cgroup 这块为性能做了什么,使用了什么高性能机制cgroup v1 / v2 的差异,以及现在的 cgroup 还有什么限制不了的cgroup 能不能描述网络带宽限制,标准实现是如何做到的手撕题:统计 1 到 n 中数字 1 出现的次数未来方向规划反问环节我的问题:您那边主要做什么业务、什么开发方向?是否涉及分布式事务?我的问题:能否给一下这次面试评价?面试官建议:横向拓展:不只是会实现 cgroup,还要理解其设计初衷、工业界诉求及未来发展;关注资源限制能力的准确性、隐患及 overhead;思考做完后如何评价其好在哪。表达呈现:认为做得比较扎实,但表达和呈现还有优化空间,需要把做过的东西讲得更结构化。
查看18道真题和解析
点赞 评论 收藏
分享
评论
1
6
分享

创作者周榜

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