Senior Machine Learning Engineer – Predictive World Model
XPeng MotorsElectric Vehicles company
Santa Clara, United States$174,720 - $295,680Senior
Coatue
Alibaba
Foxconn
IDG Capital
Xiaomi
Aspex
Data & AI
About the role
TL;DR
Develop predictive world models for autonomous driving and robotics using generative AI.
- •XPENG is seeking Machine Learning Engineers with strong expertise in generative modeling and large-scale deep learning systems to research, implement, and evaluate world models.
- •These models will learn the dynamics of the physical world from multimodal data, serving as learned simulators for training and evaluating driving and robotic policies.
- •Key Responsibilities Research and develop predictive world models that forecast the future state of a scene from large-scale multimodal driving and robotics data.
- •Develop high-quality multi-view future prediction and generation.
- •Work at the boundary between world modeling and policy learning.
- •Extend prediction beyond 2D pixel into a shared multimodal latent space.
- •Advance cross-embodiment generalization for world models.
- •Requirements MS or PhD level education in Engineering or Computer Science with a focus on Deep Learning, Computer Vision, Generative Models, or a related field, or equivalent experience.
- •Strong experience in applied deep learning including model architecture design, large-scale model training, data curation, and empirical analysis. 1-3 years + of experience working with DL frameworks such as PyTorch, including hands-on experience with distributed training.
- •Strong Python programming experience with software design skills.
- •Solid understanding of data structures, algorithms, code optimization and large-scale data processing.
Required skills
PythonPyTorchComputer VisionLLMs
Domain expertise
automotiveai
Benefits & perks
bonus, equity, benefits, Infrastructures and computational resources, Competitive compensation package, Snacks, lunches, dinners, and fun activities
Tech stack
PythonPyTorchComputer VisionLLMs