提示:国家大学生就业服务平台提醒您注意甄别违法、虚假、高风险招聘信息,警惕索要隐私信息的行为。在投递简历前务必提前确认企业资质与职位真实性,若遇到诈骗、泄露个人隐私等情况,请向我们投诉举报。
- AI infra工程师
- [全职]
- ——
- AI infra工程师
- 收藏
- 举报
- 面议
- |
- 硕士及以上
- |
- 招聘 10人
- 09-10 18:00 更新
专业不限
来源:
南开大学
wuk84i2f8m3z3bj4ftxi5zbafoym52dm
-
天津市市辖区
职位详情
岗位职责:
1.适配主流推理和训练框架,如Megatron/vllm/transformers等;
2.基于公司自研GPU芯片,适配并优化热点算子,如flash-attn等;
3.开发调试工具,如性能分析器、可视化工具,辅助AI模型部署与调优;
4.支持PyTorch/TensorFlow等框架的模型转换与量化编译;
5.分布式推理框架如vllm/sglang部署,熟悉DP/PP/TP/EP等并行机制,调优模型推理性能。
任职要求:
1.计算机等相关专业,硕士及以上学历;
2.熟悉LLVM/MLIR/TVM等编译器框架者优先;
3.了解深度学习模型(CNN/Transformer等)和计算图优化技术(如算子融合、量化、剪枝);
4.熟悉编译原理(如IR设计、代码生成、循环优化);
5.对计算机体系结构有基本了解(如CPU/GPU/NPU架构、内存层次结构);
6.有CUDA/OpenCL等异构编程开发经验者加分;
7.熟练掌握C++/Python,有较强的算法实现能力。
1.适配主流推理和训练框架,如Megatron/vllm/transformers等;
2.基于公司自研GPU芯片,适配并优化热点算子,如flash-attn等;
3.开发调试工具,如性能分析器、可视化工具,辅助AI模型部署与调优;
4.支持PyTorch/TensorFlow等框架的模型转换与量化编译;
5.分布式推理框架如vllm/sglang部署,熟悉DP/PP/TP/EP等并行机制,调优模型推理性能。
任职要求:
1.计算机等相关专业,硕士及以上学历;
2.熟悉LLVM/MLIR/TVM等编译器框架者优先;
3.了解深度学习模型(CNN/Transformer等)和计算图优化技术(如算子融合、量化、剪枝);
4.熟悉编译原理(如IR设计、代码生成、循环优化);
5.对计算机体系结构有基本了解(如CPU/GPU/NPU架构、内存层次结构);
6.有CUDA/OpenCL等异构编程开发经验者加分;
7.熟练掌握C++/Python,有较强的算法实现能力。
看了该职位的人还会看 更多