- AI infra工程师
- [全职]
- ——
- AI infra工程师
- 收藏
- 举报
- 面议
- |
- 硕士及以上
- |
- 招聘 10人
- 09-10 18:00 更新
-
天津市市辖区
职位详情
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,有较强的算法实现能力。
看了该职位的人还会看 更多