AI infra工程师
杭州
硕士及以上
不限专业
使用简历深度优化功能,快速提升简历质量
职位介绍
1. AI框架适配:- 适配主流推理和训练框架,如Megatron/vllm/transformers等。- 基于公司自研GPU芯片,适配并优化热点算子,如flash-attn等。2. 工具链支持:- 开发调试工具,如性能分析器、可视化工具,辅助AI模型部署与调优。- 支持PyTorch/TensorFlow等框架的模型转换与量化编译。3.有分布式推理框架如vllm/sglang部署经验,熟悉DP/PP/TP/EP等并行机制,调优过模型推理性能者优先1. 编程能力:- 熟练掌握C++/Python,有较强的算法实现能力。- 熟悉LLVM/MLIR/TVM等编译器框架者优先。2. AI与编译基础:- 了解深度学习模型(CNN/Transformer等)和计算图优化技术(如算子融合、量化、剪枝)。- 熟悉编译原理(如IR设计、代码生成、循环优化)。3. 硬件基础:- 对计算机体系结构有基本了解(如CPU/GPU/NPU架构、内存层次结构)。- 有CUDA/OpenCL等异构编程开发经验者加分。4. 硕士及以上学历
所属批次:新紫光集团有限公司2027秋招 2027届校园招聘简章企业职位库在招档案→

