
英伟达
AI Computing Software Development Engineer, TensorRT
AI Computing Software Development Engineer, TensorRT
发布于 大约 2 个月前普通员工/个人贡献者
上海市
中级经验
全职员工
仅现场办公
硕士
软件工程
Performance Optimization
CUDA
GPU
LLM
PyTorch
TensorFlow
TensorRT
AI 估算 · 35k–55k
英伟达作为AI芯片巨头,上海研发岗位薪资竞争力强,硕士+2年经验,结合行业水平,月薪35-55K合理。
职位详情
关于这个职位
该职位负责开发和优化英伟达的TensorRT推理软件,用于加速深度学习模型在GPU上的运行
你将与全球顶尖的AI研究团队合作,追踪LLM、生成式AI等前沿技术,并将最新研究成果集成到产品中
适合对高性能计算和深度学习有强烈兴趣的软件工程师
最低要求
计算机工程、计算机科学、应用数学或相关计算领域的硕士或更高学位(或同等经验)
年以上相关软件开发经验
优秀的C/C++编程和软件设计技能,包括调试、性能分析和测试设计
对人工智能有强烈好奇心,了解深度学习的最新发展,如LLM、生成模型和推荐模型
有使用TensorFlow和PyTorch等深度学习框架的经验
积极主动,能够在无监督下工作
优秀的英语书面和口头沟通能力
工作职责
设计和开发稳健的推理软件,可扩展至多个平台以实现功能和性能
性能分析、优化和调优
紧跟人工智能领域的学术进展,并更新TensorRT功能
为架构和硬件设计与开发提供反馈
跨公司协作,指导机器学习推理的方向,与软件、研究和产品团队合作
在科学会议上发表关键成果
AI 洞察
优缺点分析
优点
- 英伟达是AI计算领域的领导者,TensorRT是其核心软件栈,技术影响力大
- 接触最前沿的AI模型和硬件架构,技能积累速度快
- 薪酬福利优厚,股票和奖金可观,职业生涯起点高
- 对C/C++和性能优化要求极高,技术门槛不低
- 与全球团队协作,需要较强的英语沟通能力
缺点 / 挑战
- 工作节奏快,交付压力大,需要快速学习并跟进学术进展
- 适合对AI和系统性能有浓厚兴趣、乐于挑战技术难题、愿意在快节奏环境中成长的工程师
角色解读
- 技术深耕:从TensorRT开发逐步深入GPU计算和AI编译器领域,成为系统优化专家
- 架构转型:积累经验后转向AI芯片架构设计或深度学习框架研发
- 管理路线:带领小团队负责特定模块或项目,向技术管理岗位发展
- 开发并优化TensorRT推理引擎,确保深度学习模型在英伟达GPU上高效运行
- 进行性能分析和调优,识别瓶颈并改进软件架构
- 跟踪学术界最新AI进展,将LLM、生成式模型等新技术集成到TensorRT中
- 与硬件团队协作,为新一代GPU架构提供软件反馈
- 精通C/C++编程,具备扎实的调试和性能分析能力
- 熟悉深度学习框架(如PyTorch、TensorFlow)和模型优化技术
- 了解GPU计算和CUDA编程,有高性能计算经验者优先
- 对AI前沿技术(如LLM、生成式AI)有强烈好奇心和持续学习能力
申请策略
- 关注英伟达的技术博客和GTC大会,了解公司技术方向
- 在面试中展现对AI推理性能优化的热情和独立思考能力
- 突出C/C++项目经验,尤其是性能优化、调试和系统编程相关成果
- 展示深度学习框架使用经验,如有模型推理优化或部署经历更佳
- 强调对AI前沿技术的了解,可列出跟踪的论文或参与的相关项目
- 如有开源贡献或技术博客,可体现技术热情和学习能力
- 深入学习CUDA编程和GPU架构,提高并行计算能力
- 动手实践TensorRT,尝试优化一个模型推理过程
面试指南
- 针对性能问题:识别瓶颈(profiling)→ 分析原因 → 提出优化方案(如算法改进、内存对齐、并行化)→ 验证效果
- 针对模型部署:描述从训练框架到推理引擎的完整流程,包括模型导出、精度校准、优化选项和性能测试
- 针对AI知识:展示对最新技术的了解,并联系到实际工程挑战(如显存占用、推理延迟)
- 请描述你如何优化一个C++程序的内存和计算性能?
- 解释TensorRT中模型校准和INT8量化的原理?
- 如何将一个PyTorch模型导出并部署到TensorRT?
- 你对LLM推理的瓶颈有什么理解?
- 如何处理多线程环境下共享资源的性能竞争?
职位点评
74
综合评分
英伟达TensorRT工程师,技术前沿、薪资高,但工作强度大,WLB一般。
从薪资福利、成长空间、工作节奏和岗位方向综合评估,方便横向比较。
更适合这类人
适合追求技术深度和前沿发展的求职者,愿意为职业成长接受高强度工作。
表现最好
成长发展
相对薄弱
工作生活
薪资福利80
成长发展95
工作生活40
使命价值70
薪资福利
80较高
英伟达薪酬在业内顶尖,福利完善,但JD未列出具体数字,属于面议。综合判断补偿性满足较好。
薪资信号面议 (35K-55K/月)
成长发展
95较高
TensorRT是AI基础设施核心,可深度接触前沿技术和学术成果,成长空间极大。
技术前沿前沿/新兴技术
技术栈TensorRT、CUDA、LLM、Generative AI、Deep Learning、Performance Optimization
业务类型profit_center
工作生活
40较低
JD未提及远程或弹性工作,明确要求快节奏交付团队,生活化满足有限。
工作模式未明确
办公地点市区核心地段
加班情况明确要求弹性/高强度
使命价值
70中等
AI基础设施工作对技术推动有积极意义,但商业色彩较浓,使命感中等。
行业发展高速增长赛道
社会影响中性/一般
创新程度开拓性创新(行业首创)
英伟达 的其他在招职位
Infrastructure Software Engineer, Deep Learning Libraries
英伟达 · 上海市AI 估算 · 35k-55kNCX Engineer, AI Accelerator
英伟达 · 上海市AI 估算 · 40k-70kSenior Autonomous Driving Software Engineer, L4 Planning
英伟达 · 北京市AI 估算 · 35k-60kSystem Co-Design Manager
英伟达 · 上海市AI 估算 · 50k-80kSenior Software Engineer - Networking
英伟达 · 上海市AI 估算 · 35k-55k
相似职位推荐
Watch Jobs