greenhouse
Senior Staff Machine Learning Engineer – Autonomous Driving Foundation Models
XPENG
CAN
Senior · 5–8 years experience
USD 244140-413160 / year
Discovered 2 weeks ago
pytorch
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
pytorch
Key Skills for This Role
pytorch
Full Job Posting
- Architectural Leadership: Lead the design of end-to-end VLA architectures, bridging multi-modal perception with high-level linguistic reasoning and precise action generation.
- World Model Development: Drive R&D in generative world models (latent dynamics) to create high-fidelity, controllable driving simulations for closed-loop training and evaluation.
- Policy Evolution: Apply Advanced RL (Online/Offline) and IL to refine driving policies, focusing on long-horizon planning and complex multi-agent interactions.
- Scaling & Data Strategy: Define scaling laws for driving foundation models, overseeing data curation, automated labeling, and post-training at a multi-billion parameter scale.
- Global Generalization: Lead the model’s adaptation strategy for overseas road conditions, ensuring robust performance across varying traffic laws and driving cultures.
- 5-8 years of expertise in Deep Learning, with a significant track record in VLM, VLA, or Embodied AI.
- Proven experience in training and deploying Foundation Models (Transformers, LLMs) at scale.
- Deep understanding of Sequential Decision Making, World Models, or Policy Gradient methods.
- Mastery of PyTorch and expertise in distributed training (DeepSpeed, Megatron, etc.).
- A "Product-First" mindset: The ability to balance cutting-edge research with the deterministic requirements of L4 production vehicles.
- A fun, supportive and engaging environment
- Infrastructures and computational resources to support your ML model development/research.
- Opportunity to work on cutting edge technologies with the top talent in the field.
- Opportunity to make significant impact on transportation revolution by the means of advancing autonomous driving
- Competitive compensation package
- Snacks, lunches, dinners, and fun activities
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at XPENG
Human Resources Manager
Brazil, USA
Discovered 2 days ago
Marketing Executive
Dubai, UAE
Discovered 1 weeks agoFull-time
Senior Staff Physical AI Data Algorithm Engineer
, CAN
Discovered 2 weeks ago
Staff Machine Learning Engineer - LLM Quantization & Deployment
, CAN
Discovered 2 weeks ago
Senior Machine Learning Engineer – Predictive World Model
, CAN
Discovered 2 weeks ago
Senior Machine Learning Engineer - Foundation Model
, CAN
Discovered 2 weeks ago
Staff Robotics Engineer / Tech Lead – Whole-Body Control & Robot Learning
, CAN
Discovered 2 weeks ago
Senior Staff AI Data Infrastructure/Pipeline Engineer
, CAN
Discovered 2 weeks ago