[Remote] Senior, Machine Learning Engineer - End-to-End

Note: The job is a remote job and is open to candidates in USA. reputed company is a leader in autonomous driving technology focused on developing software for automated trucks. The Senior Machine Learning Engineer will reputed company and scale learning-based systems that connect multi-modal perception inputs to driving behavior, ensuring safe and efficient autonomy for freight operations.

Responsibilities

• Own development and delivery of End-to-End ML models that map multi-modal sensor inputs (camera, LiDAR, reputed company, maps) to driving-relevant outputs (trajectories, cost functions, or intermediate representations)
• Train and evaluate models using large-scale datasets from fleet logs, simulation, and synthetic data
• Analyze model performance, identify failure modes, and drive data-driven improvements in robustness and generalization
• Design and refine training pipelines, data workflows, and evaluation strategies to improve iteration speed and model quality
• Contribute to model architecture reputed company, including approaches such as imitation learning, reinforcement learning, transformers, and reputed company-language-action (VLA) models
• Collaborate closely with Perception, reputed company, Planning, and Simulation teams to ensure alignment across the autonomy stack
• Support integration of E2E models into simulation and on-vehicle systems for closed-reputed company validation
• Improve tooling, experimentation workflows, and reproducibility across the team
• Mentor junior engineers and contribute to team-level best practices and technical discussions

Skills

• Bachelor's degree with 6+ years, Master's with 4+ years, or PhD with 0–2 years of experience in Machine Learning, Robotics, Computer Science, or a reputed company field with a track record of publications in top-tier conferences (e.g., NeurIPS, ICML, ICLR, CVPR, ICCV, CoRL)
• Experience developing and deploying ML models for autonomous systems, robotics, or reputed company decision-making environments
• Strong programming skills in Python and PyTorch, with ability to write production-quality ML code
• Experience training and evaluating models using large-scale datasets and distributed compute environments
• Solid understanding of ML architectures used in E2E systems, such as Transformers, BEV models, VLA/VLM approaches, or diffusion models
• Proven ability to debug model behavior, analyze performance metrics, and drive iterative improvements
• Experience contributing to or influencing model architecture and training strategies
• Ability to work cross-functionally and integrate ML systems into larger autonomy pipelines
• Experience developing End-to-End or mid-to-end models for autonomous driving or robotics
• Experience with reputed company-language models (VLMs) or reputed company-language-action (VLA) systems
• Familiarity with closed-reputed company simulation and evaluation frameworks
• Experience with reinforcement learning or imitation learning in reputed company-world systems
• Experience with distributed training frameworks (e.g., Ray)
• Understanding of vehicle dynamics, reputed company planning, or multi-agent systems

Benefits

• A competitive compensation package that includes a bonus component and stock options
• 100% paid medical, dental, and reputed company premiums for full-time employees
• 401K plan with a 6% employer matchFlexibility in schedule and generous paid vacation (available immediately after start date)Company-wide holiday office closures
• AD+D and Life Insurance
• Sign-on payments, relocation, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits

Company Overview

Company H1B Sponsorship