This job has expired

This job posting is no longer active and is not accepting applications. Explore similar roles below!

PlusAI
Posted 2mo ago

Machine Learning Engineer Intern- Scenario Simulation

PlusAI
Santa Clara, California, United States
$19-$65/hrOnsiteInternship
Responsibilities
  • Developing BEV embeddings
  • Optimizing simulation throughput
  • Enabling closed-loop training
Requirements
  • Strong foundation in deep learning
  • Computer vision
  • Python, and PyTorch
  • Experience with BEV/E2E autonomous driving concepts
  • Familiarity with simulation tools and data generation
Technical tools mentioned
PythonPyTorchBEVLidarCARLAKubeflowMLFlowWeights & BiasesC++

Job description

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.

This project addresses the sim-to-real gap as in our current simulator pipeline. Closed-loop simulation systems are highly effective at generating massive amounts of diverse, high-fidelity ground truth data. However, they assume perfect perception. However, this ideal GT data does not perfectly align with the BEV feature representations extracted from real-world, noisy sensor data.

The project provides two outcomes:
Bridge Sim-to-Real Gap in Data Augmentation with Closed-Loop Simulation: By translating simulation GTs into a realistic BEV embedding space, the synthetic data becomes directly usable for training and fine-tuning BEV-based planning models.
Allow BEV Space Planning Model to be directly fine-tunable via Self-play RL: Once the sim-to-real gap is bridged in the BEV feature space, the planning models can thus be directly finetuned / post-trained in a self-play RL setting.


In this role you will work with the Scenario Simulation team in enhancing the fidelity of PLUS’ in-house simulator, with a focus on reducing the gap between simulated and real-world behavior. This includes improving the robustness of log-replay simulations, enhancing log conversion methods, and implementing features such as warm-start for varied initial conditions. You will also contribute to synthetic scenario development, including features like intelligent traffic light control, construction zone creation, and realistic agent spawning for local and highway environments. The work will help improve simulation frameworks used to validate and deploy autonomous truck systems.


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.

    About PlusAI

    AI-based virtual driver software for factory-built autonomous trucks.

    Year founded
    2016
    Employees
    466
    Organization type
    Private
    Latest investment
    Raised $720.00M Series C (2021) — led by FountainVest Partners, ClearVue Partners, CPE, Guotai Junan International
    Headquarters
    US

    Similar jobs

    Machine Learning Engineer roles near Santa Clara, California
    18h
    Save
    Mark Applied
    Hide
    Machine Learning Engineer
    South San Francisco, California, United States
    $212k-$318k/yr HybridFull Time
    Stripe
    Stripe: Provides online payment processing and financial infrastructure for businesses.
    2+ YOEBachelor's degree in a relevant quantitative or computing field and 2+ years building, deploying, and operating production ML systems, including model development, data pipelines, and large-scale deployment.
    PyTorch, TensorFlow, XGBoost, Apache Spark
    19h
    Save
    Mark Applied
    Hide
    Machine Learning Engineer, Tapestry
    Mountain View or United States or United Kingdom or Chile or New Zealand or Australia or Brazil
    $166k-$244k/yr HybridFull Time
    X, The Moonshot Factory
    X, The Moonshot FactoryNASDAQ: GOOGL: Develops early-stage technologies and experimental projects.
    Bachelor's or master's degree in ML, computer science, statistics, or related field; ML engineering experience; Python and PyTorch or TensorFlow; expertise in specified ML domains.
    Python, PyTorch, TensorFlow, AWS, GCP, Azure
    20h
    Save
    Mark Applied
    Hide
    Senior Staff Machine Learning Engineer - Agentic AI
    Santa Clara, California, United States
    $201k-$352k/yr OnsiteFull Time
    ServiceNow
    ServiceNowNYSE: NOW: Provides a cloud platform for automating enterprise digital workflows.
    8+ YOERequires 8+ years of software engineering, production agentic AI experience, machine learning fundamentals, Python, distributed systems, and technical leadership. Go, Java, or C++ and frontier AI SDK experience are required or preferred.
    Python, Go, Java, C++, Anthropic, Google, OpenAI, Anthropic's Claude Cowork, MCP, NVIDIA, Cursor, Claude Code, GitHub Copilot, CMDB, Workflow Data Fabric, Knowledge Graph, RAG, RAG pipelines, hybrid search, vector stores, MLOps
    20h
    Save
    Mark Applied
    Hide
    Senior Staff Machine Learning Engineer - Agentic AI
    Santa Clara, California, United States
    $201k-$352k/yr OnsiteFull Time
    ServiceNow
    ServiceNowNYSE: NOW: Enterprise cloud platform for digital workflow automation.
    8+ YOERequires 8+ years of software engineering experience, production agentic AI expertise, Python, distributed systems, machine learning fundamentals, frontier AI SDKs, and technical leadership.
    Python, Go, Java, C++, Anthropic, Google, OpenAI, CMDB, Workflow Data Fabric, Knowledge Graph, RAG, MCP, NVIDIA, Cursor, Claude Code, GitHub Copilot
    22h
    Save
    Mark Applied
    Hide
    Senior Machine Learning Engineer, Infrastructure
    New York City or San Francisco
    $212k-$318k/yr HybridFull Time
    Patreon
    Patreon: Membership platform for creators to monetize work and engage fans.
    Deep experience building production ML infrastructure, low-latency inference pipelines, feature stores, distributed systems, backend engineering, Python, debugging, documentation, and code reviews.
    Python
    1d
    Save
    Mark Applied
    Hide
    Machine Learning Engineer Graduate (E-Commerce Governance) - 2027 Start
    San Jose, California, United States
    $128k-$317k/yr OnsiteFull Time
    TikTok
    TikTok: Global short-form video hosting and social media platform.
    Bachelor's degree in software development, computer science, computer engineering, or related field; strong Python, Go, or C++ skills; Linux development; data structures, algorithms, machine learning, and deep learning knowledge.
    Linux, Python, Go, C++, natural language processing, computer vision, multimodal, graph algorithms, search algorithms, text mining, data mining, LLM, reinforcement learning, operational research
    1d
    Save
    Mark Applied
    Hide
    Principal Machine Learning Engineer
    San Jose or Seattle or San Francisco
    $206k-$379k/yr OnsiteFull Time
    Adobe
    AdobeNASDAQ: ADBE: Provides software for digital media creation and marketing analytics
    8+ YOERequires 8+ years in production-scale machine learning engineering, 3+ years leading GPU-intensive GenAI inference systems, and expertise in inference tools, APIs, distributed systems, and technical leadership.
    PyTorch, CUDA, Triton, TensorRT, Nvidia Dynamo, Python, Kubernetes, MLOps
    1d
    Save
    Mark Applied
    Hide
    Machine Learning Engineer II
    San Jose or Seattle
    $119k-$181k/yr HybridFull Time
    Expedia Group
    Expedia GroupNASDAQ: EXPE: Operates a global platform for travel bookings and services.
    2+ YOEBachelor’s degree or equivalent, 2+ years of machine learning engineering experience, Python or Java, ML frameworks, production deployment, and real-time ML systems experience.
    Python, Java, Spark, Databricks, AWS, CI/CD
    This job has expired