Senior ML Engineer, AI Foundation for Large Models

XPENG; Santa Clara (CA); On-site; USD 174,720 - 295,680; Full time

Benefits offered by this job

Competitive compensation package

Supportive work environment

Snacks, lunches, and fun activities

Job summary

XPENG is seeking a full-time Machine Learning Engineer - AI Foundation in Santa Clara, California. The role involves optimizing transformer models for autonomous driving solutions while working alongside a talented team.

Ideal candidates will have a Master's in Computer Science or equivalent, along with strong expertise in PyTorch, programming in Python and C++. Competitive compensation and a supportive work environment are provided.

Qualifications

• 3+ years of industry experience in machine learning or software engineering.
• Strong knowledge of optimizing transformer models.
• Experience with deployment across diverse hardware.

Responsibilities

• Optimize transformer-based LLMs for low-latency and high-throughput inference.
• Deploy optimized models across GPUs, CPUs, and edge accelerators.
• Contribute to internal tooling and documentation for model optimization.

Skills

PyTorch

Programming in Python

Programming in C++

Transformer architecture knowledge

Education

Master in CS/CE/EE or equivalent

Tools

CUDA

Triton

Torchscript

Nvidia TensorRT

Job description

XPENG is seeking a full-time Machine Learning Engineer - AI Foundation in Santa Clara, California. The role involves optimizing transformer models for autonomous driving solutions while working alongside a talented team.

Ideal candidates will have a Master's in Computer Science or equivalent, along with strong expertise in PyTorch, programming in Python and C++. Competitive compensation and a supportive work environment are provided.