越南市场交付/解决方案
2026-06-22
智元创新（上海）科技股份有限公司· 巴亭·
岗位详情
加入投递
智
优才-模型推理AI Infra开发工程师-中央研发部
智元创新（上海）科技股份有限公司| 上海/北京
校招本科
发布于 2026-06-17
职位描述
1. 设计面向具身机器人本体亲和的轻量化、高性能推理框架，支持CNN、Transformer、Mamba、GNN等主流网络结构；
2. 面向GPU/NPU的具身多模态算子开发与性能优化，实现内存复用、算子融合、数据排布优化、多流并行等通用加速策略;
3.对模型推理过程进行时间、功耗、内存占用分析，定位瓶颈并设计软硬协同优化策略；
4. 实现支持 PTQ/QAT 的量化工具链(INT8/INT4/FP8/MXFP8等浮点混合精度)；
5. 跟进VLA、世界模型、空间智能等最新算法架构趋势，及时洞察捕捉低精度量化、AI KernelGen、投机推理等在具身本体上的模型高效推理适配技术
任职要求
1. 熟练掌握C/C++/Python/Go至少一种编程语言，具备CUDA/OpenCL/CANN优化经验；
2. 深入理解至少一种推理引擎架构(如tensorrt/vllm/sglang/ktransformer/llama-cpp)推理框架架构；
3. 熟悉模型量化原理(KL 散度、MinMax、LSQ、AWQ、GPTQ等)及相关工具链；
4. 熟悉常见典型算子实现与优化(卷积、矩阵乘、激活、LayerNorm、Softmax、flashattention等)；
5. 熟悉MLIR、IREE、Triton、TileLang等至少一中AI编译框架