AI Large Model Acceleration Optimization Algorithm Engineer
- Xiaomi
- Shanghai, Beijing, China
- CNY 300,000 – CNY 600,000
北京、上海社招全职职位 ID：A7018
职位描述
岗位职责： 1.进行大模型、多模态大模型、AIGC模型、自驾模型等多个方向端侧加速优化算法的研发与探索 2.工作内容包括但不限于上述模型的训练调优、量化、剪枝、小型化、稀疏优化、投机推理、专用加速、结构搜索（NAS）等 3.与芯片团队配合针对相关加速特性进行芯片算法联合设计与优化，并提供针对自研芯片的完整算法加速解决方案 4.帮助客户在自研芯片上进行算法加速优化的模型加速优化训练及交付
职位要求
岗位要求： 1.从事大模型、AIGC、深度学习、AI芯片等两个或以上相关领域工作经验，有2年及以上的工作经验，主导或者作为主要参与者； 2.对AI算法（尤其是大模型、AIGC、AD算法）的量化剪枝、网络结构搜索、稀疏化、性能效果优化和训练调参等有丰富开发调试经验； 3.精通transfomers，diffuser，pytorch，tensorrt等框架或平台； 4.参加过完整芯片算法联合优化特性的设计和商用落地者优先； 5.有大模型/AIGC/AD算法加速优化项目经验者或在相关领域发表过学术论文或有开源项目者优先；
投递
Skills
- PyTorch
- TensorRT
- Model Quantization
- Model Pruning
- Neural Architecture Search
- Deep Learning
- Large Language Models
