jobapplier

Senior/Staff Research Engineer (Vision-Language-Action Models, Autonomous Driving)

PlusAI

Skills

Required

  • model training
  • implementing and training models
  • PyTorch / TensorFlow / JAX
  • diffusion
  • flow matching
  • transformers

Preferred

  • model training, evaluation
  • SFT
  • RL
  • post-training
  • large multimodal models
  • multi-modal sensor data
  • camera
  • LiDAR

Join our core AI team as a Senior/Staff Research Engineer, where you will be at the forefront of autonomous decision-making. You will design, train, and evaluate Vision-Language-Action models that generate high-level driving decisions and trajectory guidance. You will own a VLA workstream end to end, build training and evaluation pipelines, and collaborate with various teams to bring models from research to production. Enjoy benefits such as unlimited PTO, flexible working arrangements, health benefits, equity options, professional development opportunities, and catered lunches.

Responsibilities

  • Conception, training, and evaluation of Vision-Language-Action models for high-level driving decisions.
  • Ownership of a VLA workstream from data collection to on-vehicle validation, including architecture and large-scale training.
  • Collaboration with cross-functional teams to transition models from research to production, ensuring rigorous performance metrics.
  • Direct, hands-on experience training vision-language / vision-language-action models
  • Model distillation, quantization, and inference optimization (ONNX/TensorRT, mixed precision, custom kernels)

Requirements

  • M.S. minimum, Ph.D. preferred in CS, EE, Mathematics, Statistics, or a related field
  • 3+ years implementing and training models in a deep learning framework (PyTorch, TensorFlow, or JAX)
  • Thorough understanding of state-of-the-art vision-language / VLA models, diffusion, flow matching, and transformers
  • Experience with large-scale / distributed model training

Preferred

  • Hands-on experience with model training, evaluation, and deployment in production

SFT and RL post-training of large multimodal models

  • Hands-on experience with multi-modal sensor data (camera, LiDAR, radar)
  • Publications at top venues (CVPR, NeurIPS, ICML, ICLR, CoRL, RSS, ICRA)
  • Autonomous driving / ADAS experience

Pay

USD 170,000 - 260,000