Senior Machine Learning Engineer - AI Foundation

XPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take-off and landing (eVTOL) aircraft, and robotics. We are looking for a full-time Machine Learning Engineer - AI Foundation, with deep knowledge and strong enthusiasm towards establishing a state-of-art ML infrastructure for training very large foundation model and accelerating model training/inference.

Requirements

• Optimize transformer-based LLMs for low-latency and high-throughput inference.
• Optimize kernels and model graphs using tools like CUDA, Triton, and custom fused operators.
• Implement and benchmark (Quantization, Knowledge distillation, structured and unstructured pruning, KV-cache optimization, etc.).
• Deploy optimized models across GPUs, CPUs, and edge accelerators.
• Contribute to internal tooling and documentation for model optimization flows.

Benefits

• Competitive compensation package.
• Snacks, lunches, dinners, and fun activities.