Senior Machine Learning Engineer - Foundation Model

XPENG
Santa Clara, CA
Category Research
Job Description
XPENG is seeking a full-time Machine Learning Engineer to establish and maintain a state-of-the-art machine learning infrastructure for training large foundation models and accelerating model training/inference. This role involves designing, training, and deploying deep learning models using PyTorch, transformer architecture, and distributed training frameworks. The engineer will contribute to the advancement of autonomous driving technology and solve challenging problems in the AI and machine learning space.

Requirements

  • Design, train, and deploy large deep learning models.
  • Use PyTorch and PyTorch Lightning.
  • Utilize transformers and accelerate model training/inference.
  • Experience training large-scale vision or language models.

Benefits

  • Competitive compensation package
  • Snacks, lunches, dinners, and fun activities
]]>