Xpengmotors
Senior Machine Learning Engineer - AI Foundation
Santa Clara, CA · Senior · Full-time
Sponsorship not specified$175k-$296kDetected 57 days ago
PythonC++Machine LearningNLPLLMsSignal ProcessingResearch
About the role
- With a strong focus on intelligent mobility, XPENG is dedicated to reshaping the future of transportation through cutting-edge R&D in AI, machine learning, and smart connectivity.
- Our salary ranges are determined by role, level, and location.
- The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all US locations.
Responsibilities
- 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.).
- Infrastructures and computational resources to support your work.
Requirements
- Good knowledge of PyTorch.
- Knowledge of transformer architecture and ways to accelerate the training and inference of transformer models.
- Previous experience in the autonomous driving industry.
- Knowledge of Torchscript and Nvidia TensorRT.
- Familiarity with GPU CPU, NPU, DSP architecture.
Compensation
- Competitive compensation package.
Benefits
- Opportunity to work on cutting edge technologies with the top talents in the field.
Apply directly at Xpengmotors →Create a free account for alerts like thisView Xpengmotors immigration profile
This listing is sourced directly from Xpengmotors's careers page and normalized into a canonical job model.