Senior/Staff Research Engineer — Vision-Language-Action Models (Autonomous Driving)
Required skills
- Vision-Language-Action Models
- VLA models
- SuperDrive
- reasoning layer
- trajectory guidance
- strategic decision-making
- model compression
- on-board compute
- PyTorch
- TensorFlow
Nice to have
- model distillation
- quantization
- inference optimization
- ONNX
- TensorRT
- mixed precision
Role summary
PlusAI is seeking a Senior/Staff Research Engineer to join its core AI team, focusing on Vision-Language-Action (VLA) models for autonomous driving. The role involves training VLA models for strategic decision-making and trajectory guidance, and designing techniques for model compression and deployment on onboard compute. Candidates should have an M.S. or Ph.D. in a related field, 3+ years of deep learning framework experience (PyTorch, TensorFlow, JAX), and hands-on experience with VLA models, training, evaluation, and production deployment. Expertise in transformers, diffusion, flow matching, and large-scale training is required. Preferred qualifications include experience with model distillation, quantization, inference optimization, multimodal sensor data, and publications at top venues.
You will join our core AI team at the frontier of autonomous decision-making, building the Vision-Language-Action (VLA) models that form SuperDrive's reasoning layer. You'll train VLA models that generate high-level driving decisions and trajectory guidance for on-board strategic decision-making, and design the knowledge distillation and compression techniques that transition large models onto on-board compute.
Responsibilities
Required qualifications
- 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).
- Direct, hands-on experience training vision-language / vision-language-action models.
- Hands-on experience with model training, evaluation, and deployment in production.
- Thorough understanding of state-of-the-art vision-language / VLA models, diffusion, flow matching, and transformers.
- Experience with large-scale / distributed model training.
Preferred Qualifications
Your opportunities joining PlusAIWork, learn and grow in a highly future-oriented, innovative and dynamic field.Wide range of opportunities for personal and professional development.Catered free lunch, unlimited snacks and beverages.Highly competitive salary and benefits package, including 401(k) plan.
Similar roles
- PData Engineer (SE / Sr SE)PlusAI · Santa Clara, CA · Hybrid
- PFleet ManagerPlusAI · Fremont, CA · Onsite
Senior AI Research EngineerPfizer · San Diego, CA · Hybrid- Staff AI Research EngineerDuolingo · New York, NY · Onsite
- Research Engineer, Model Evaluation and ImprovementBenchling · San Francisco, CA · Onsite
Research EngineerKlavis AI · Remote · Remote
