AI推理框架开发工程师/专家
- Xiaomi
- Shanghai, Beijing +1 more, China
- CNY 500,000 – CNY 900,000
上海、北京、西安社招全职职位 ID：K9595
职位描述
基于AI芯片，完成AI推理框架的方案设计及开发，包括： 1.基于AI芯片，完成 AI 推理框架的开发和交付，适配CPU/GPU/NPU多后端，支持CV/NLP/大模型等端侧落地； 2.负责框架与FrameWork的对接，包括AndroidNN/Tensorflow/Pytorch/Onnx等； 3.负责框架相关工具链的开发，包括Memory/PerformanceProfiling，量化工具等； 4.负责CV/NLP/大模型等算法业务的端到端性能优化，满足商用交付需求
职位要求
1.硕士及以上学历，计算机/电子/信息/通信/自动化/软件等相关专业，有AI工程优化经验的优先； 2.精通C/C++，精通算法与数据结构，熟悉Python； 3.深入了解PADDLE/MNN/TVM/MLIR等框架中的一种或多种，熟悉算子性能优化、算子融合、graph遍历等技术； 4.熟悉NPU或其他AI芯片工具链开发， 熟悉模型量化技术优先； 5.了解主流 AIGC 算法模型原理，有AIGC模型加速优化经验者优先。 6.熟悉Android NN/TensorFLow/Pytorch/Caffe中一种神经网络框架；
投递
Skills
- C/C++
- AI Inference Framework Development
- NPU Toolchain Development
- Model Quantization
- Performance Optimization
- Python
- Deep Learning Frameworks (PyTorch/TensorFlow)
