Staff Machine Learning Engineer - Foundation Model

Key Responsibilities: Design and implementlarge‑scale multi‑modal architectures(e.g., vision–language–action transformers) for end‑to‑end autonomous driving. Developpretraining and fine‑tuning strategiesleveraging massive labeled and unlabeled fleet data (images, video, LiDAR, CAN bus, maps, human driving behaviors, etc.).Research and integratecross‑modal alignment(e.g., visual grounding, temporal reasoning, policy distillation, imitation and reinforcement learning) to improve model interpretability and action quality. Collaborate with infrastructure engineers toscale training across thousands of GPUsusing distributed training frameworks (FSDP, DDP, etc.).Conductsystematic ablation, evaluation, and visualizationof model behavior across perception, reasoning, and planning tasks. Contribute tomodel deploymentoptimization, including quantization, export, and latency–accuracy trade‑offs for onboard execution.

Minimum Qualifications: Master’s degree or higher inComputer Science, Electrical/Computer Engineering, or related field, with 3+ years of experiencein deep learning research or productization. Strong proficiency inPyTorchand modern transformer‑based model design. Experience inlarge‑scale pretrainingormulti‑modal modeling(vision, language, or planning).Deep understanding ofrepresentation learning, temporal modeling, andself‑supervised orreinforcement learningtechniques. Familiarity withdistributed training(DDP, FSDP) and large‑batch optimization.

Preferred Qualifications: PhD inCS/CE/EEor related field, with 1+ years of relevant industry experience. Publication record in top‑tier AI conferences (CVPR, ICCV, NeurIPS, ICLR, ICML, ECCV).Prior experience buildingfoundation or end‑to‑end driving models, orLLM/VLM architectures(e.g., ViT, Flamingo, BEVFormer, RT‑2, or GRPO‑style policies).Familiarity withRLHF/DPO/GRPO , trajectory prediction, orpolicy learningfor control tasks. Proven ability to collaborate cross‑functionally with infra, perception, and planning teams to deliver production‑ready models. What do we provide: A collaborative, research‑driven environment with access tomassive real‑world dataandindustry‑scale compute. An opportunity to work withtop‑tier researchers and engineersadvancing the frontier of foundation models for autonomous driving. Direct impact on the next generation ofintelligent mobility systems .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. The base salary range for this full‑time position is $215,280-$364,320, in addition to bonus, equity and benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job‑related skills, experience, and relevant education or training. We are an Equal Opportunity Employer. It is our policy to provide equal employment opportunities to all qualified persons without regard to race, age, color, sex, sexual orientation, religion, national origin, disability, veteran status or marital status or any other prescribed category set forth in federal or state regulations.