Job Board
LogoLogo

Get Jobs Tailored to Your Resume

Filtr uses AI to scan 1000+ jobs and finds postings that perfectly matches your resume

XPeng Motors Logo

Staff Machine Learning Engineer - AI Foundation

XPeng Motors

$215,280 - $364,320
Nov 14, 2025
Santa Clara, CA, US
Apply Now

Solving the autonomous driving problem through cutting-edge R&D in AI, machine learning, and smart connectivity.

Requirements

  • Deep knowledge of PyTorch.
  • Knowledge of model training framework (e.g. PyTorch Lightning)
  • In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models.
  • Experience of performing large scale distributed training of models.
  • Knowledge of Nvidia TensorRT.
  • Experience with cuda or Triton language for writing custom ops.
  • Experience with edge computing systems.

Responsibilities

  • Design, train, and deploy large deep learning models that can leverage the vast amount of labeled and unlabeled data from a fleet of million vehicles.

Other

  • PhD in CS/CE/EE, or equivalent, in industry experience.
  • A fun, supportive and engaging environment
  • Infrastructures and computational resources to support your work.
  • Opportunity to work on cutting edge technologies with the top talents in the field.
  • Opportunity to make significant impact on the transportation revolution by the means of advancing autonomous driving
  • Competitive compensation package
  • Snacks, lunches, dinners, and fun activities
  • A track record of efficiently solving complex problems collaboratively on larger teams