直招.cv

← 返回职位列表

小米 实习

端到端算法工程师实习生

北京 ·经验不限·学历不限

北京

职位描述

1. Develop quantization, sparsity, pruning, and distillation techniques to enhance the production-level autonomous driving models. 2. Optimize, convert, and deploy autonomous driving models (e.g., ONNX models) that operate efficiently on diverse hardware (GPU, CPU, in-house AI ASIC). 3. Design and implement custom kernels using C++ and CUDA to accelerate model operations and pre/post-processing pipelines. 4. Perform a systematic benchmarking, scaling, and validation of inference performance across various hardware platforms (GPU, CPU, in-house AI ASIC). 5. Collaborate with hardware, compiler, and AI infra engineers to achieve efficient and accurate AI model inference. 【任职要求】 1. Familiar with techniques like quantization, sparsity, pruning, and distillation for edge and real-time inference. 2. Strong proficiency in Python and C++, and deep learning frameworks such as PyTorch. 3. Hands-on expertise with CUDA programming, low-level performance profiling, and compiler-level optimization. 4. Strong understanding of computer systems and architecture, with experience deploying AI models on GPUs and NPUs. 5. Experience with model optimization tools like ModelOpt/ONNXSim/TensorRT, as well as deploying models on edge devices or mobile platforms. 6. Strong problem-solving skills with the ability to debug and optimize high-performance inference workloads. 7. Experience collaborating with hardware/compiler/AI infra engineers to connect model-level and system-level optimization.

官网发布:2026-04-07 · 最后确认在招:2026-09-03 23:17:41 · 来源平台:xiaomi