Machine Learning Engineer Intern - Scenario Simulation

PlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. Headquartered in Silicon Valley with operations in the United States and Europe, Plus was named by Fast Company as one of the World’s Most Innovative Companies. Partners including TRATON GROUP’s Scania, MAN, and International brands, Hyundai Motor Company, Iveco Group, Bosch, and DSV are working with Plus to accelerate the deployment of next-generation autonomous trucks. If you’re ready to make a huge impact and drive the future of autonomy, Plus is looking for talented individuals to join its fast-growing teams.

Responsibilities:

Bridge the Sim-to-Real Gap: Develop and implement a translation layer that converts idealized simulator Ground Truth (GT) into realistic, "noisy" Bird’s-Eye View (BEV) embeddings.

Optimize Simulation Throughput: Research and implement a "shortcut" pipeline that bypasses slow image/LiDAR rendering to generate BEV features directly from state data.

Enable Closed-Loop Training: Integrate the translated BEV embeddings into a training pipeline to make synthetic data directly usable for planning models.

Support Reinforcement Learning: Create the infrastructure necessary for planning models to undergo self-play RL fine-tuning within the bridged BEV feature space.

Required Skills:

Strong foundation in deep learning, computer vision, and machine learning.

Proficiency in Python and deep learning frameworks (PyTorch)

Prior experience with BEV / E2E Autonomous Driving Architectures (understanding of BEV generation, sensor fusion, spatial transformation).

Prior experience in addressing the Sim-to-Real Gap in autonomous systems such as robotics and autonomous driving

Preferred Skills:

Nice-to-have:

Experience with C++

Experience with End-to-End (E2E) driving models.

Experience with working with simulators (CARLA, IsaacSim, Gazebo)

Experience with curating datasets and synthetic data generation (SDG),

Experience with ML Infras (Kubeflow, MLFlow, Weights & Biases, etc)

Basic understanding of simulation / scenario-based testing for autonomous driving systems

Your opportunities joining PlusAI

Work, 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.