<div>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.</div><br><br><h3>Responsibilities:</h3><ul><li><span>Identify Training Bottlenecks: Profile and analyze Bird's Eye View (BEV) model training pipelines to pinpoint computational and memory bottlenecks.</span></li><li><span>Develop Custom Kernels: Design and implement high-performance custom compute kernels using CUDA, Triton, or C++ to accelerate the model training process.</span></li><li><span>Leverage LLMs for Optimization: Explore and integrate Large Language Models (LLMs) to assist in generating high-performance code and optimizing kernel logic.</span></li><li><span>Automate Profiling Workflows: Build systems to automate performance profiling and analysis using tools like NVIDIA Nsight and the PyTorch Profiler.</span></li><li><span>Iterative Performance Tuning: Continuously analyze profiling data generated by both human and LLM-assisted workflows to maximize GPU utilization and reduce training times.</span></li></ul><h3>Required Skills:</h3><ul><li><span>Systems Programming: Strong proficiency in C++ and a solid understanding of memory management, computer architecture, and parallel processing principles.</span></li><li><span>Deep Learning Frameworks: Hands-on experience with PyTorch, specifically understanding custom operations, autograd, and training loops.</span></li><li><span>Performance-Oriented Mindset: Strong problem-solving skills with a deep interest in performance tuning, algorithmic efficiency, and low-level system optimization.</span></li></ul><h3>Preferred Skills:</h3><ul><li><span>GPU Programming Experience: Practical experience writing and optimizing custom GPU kernels using CUDA or OpenAI Triton.</span></li><li><span>Hardware Profiling Tools: Familiarity with hardware and software profiling tools, particularly NVIDIA Nsight (Systems/Compute) and the PyTorch Profiler.</span></li><li><span>LLM for Code Generation: Experience using or prompting LLMs for code writing, refactoring, or exploring AI-assisted software development workflows.</span></li><li><span>Autonomous Vehicle Perception: A foundational understanding of Bird's Eye View (BEV) models, 3D perception, or spatial transformers used in autonomous driving.</span></li></ul><br><div><b>Your opportunities joining PlusAI</b></div><div>Work, learn and grow in a highly future-oriented, innovative and dynamic field.</div><div>Wide range of opportunities for personal and professional development.</div><div>Catered free lunch, unlimited snacks and beverages.</div><div>Highly competitive salary and benefits package, including 401(k) plan.</div>

<div class="org-about-module__description">
<div id="ember51" class="ember-view organization-about-module__content-consistant-cards-description"><span class="lt-line-clamp__raw-line">Plus is an AI company whose mission is to build &ldquo;driving intelligence&rdquo; to power a safer and greener world. Plus&rsquo;s autonomous driving solutions span from driver-out SuperDrive&trade;, to highly automated PlusDrive&reg;, next-gen safety technology PlusProtect&trade;, and model-based perception software PlusVision&trade;. Headquartered in Silicon Valley with operations in the U.S., Europe, and Australia, Plus is named by Fast Company as one of the World&rsquo;s Most Innovative Companies. Plus&rsquo;s large AI models are already powering vehicles in commercial use today. Partners including Bosch, dm-drogerie markt, DSV, Hyundai Motor Company, IVECO, Luminar, Nikola, Scania/MAN/Navistar of the TRATON GROUP, and Transurban are working with Plus to accelerate next-generation transportation solutions.</span></div>
</div>
<div class="org-about-module__content-improvement">
<div class="org-about-module__content-box">&nbsp;</div>
</div>

Machine Learning Infrastructure Engineer Intern

<div>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.</div><br /><br /><h3>Responsibilities:</h3><ul><li><span>Identify Training Bottlenecks: Profile and analyze Bird's Eye View (BEV) model training pipelines to pinpoint computational and memory bottlenecks.</span></li><li><span>Develop Custom Kernels: Design and implement high-performance custom compute kernels using CUDA, Triton, or C++ to accelerate the model training process.</span></li><li><span>Leverage LLMs for Optimization: Explore and integrate Large Language Models (LLMs) to assist in generating high-performance code and optimizing kernel logic.</span></li><li><span>Automate Profiling Workflows: Build systems to automate performance profiling and analysis using tools like NVIDIA Nsight and the PyTorch Profiler.</span></li><li><span>Iterative Performance Tuning: Continuously analyze profiling data generated by both human and LLM-assisted workflows to maximize GPU utilization and reduce training times.</span></li></ul><h3>Required Skills:</h3><ul><li><span>Systems Programming: Strong proficiency in C++ and a solid understanding of memory management, computer architecture, and parallel processing principles.</span></li><li><span>Deep Learning Frameworks: Hands-on experience with PyTorch, specifically understanding custom operations, autograd, and training loops.</span></li><li><span>Performance-Oriented Mindset: Strong problem-solving skills with a deep interest in performance tuning, algorithmic efficiency, and low-level system optimization.</span></li></ul><h3>Preferred Skills:</h3><ul><li><span>GPU Programming Experience: Practical experience writing and optimizing custom GPU kernels using CUDA or OpenAI Triton.</span></li><li><span>Hardware Profiling Tools: Familiarity with hardware and software profiling tools, particularly NVIDIA Nsight (Systems/Compute) and the PyTorch Profiler.</span></li><li><span>LLM for Code Generation: Experience using or prompting LLMs for code writing, refactoring, or exploring AI-assisted software development workflows.</span></li><li><span>Autonomous Vehicle Perception: A foundational understanding of Bird's Eye View (BEV) models, 3D perception, or spatial transformers used in autonomous driving.</span></li></ul><br /><div><b>Your opportunities joining PlusAI</b></div><div>Work, learn and grow in a highly future-oriented, innovative and dynamic field.</div><div>Wide range of opportunities for personal and professional development.</div><div>Catered free lunch, unlimited snacks and beverages.</div><div>Highly competitive salary and benefits package, including 401(k) plan.</div>

Machine Learning Infrastructure Engineer Intern

About the Job

About the Company

Plus

Similar Jobs

Senior Machine Learning Engineer - AI Foundation

Senior Machine Learning Engineer - AI Foundation

Staff Machine Learning Engineer - AI Foundation

Staff Machine Learning Engineer - AI Foundation

AI Intern – VLA Deployment

AI Intern – VLA Deployment

Full Stack Engineer Intern

Full Stack Engineer Intern

Simulation Engineer Intern

Simulation Engineer Intern

Machine Learning Engineer Intern

Machine Learning Engineer Intern

Machine Learning Engineer Intern - Scenario Generation

Machine Learning Engineer Intern - Scenario Generation

Deep Learning Intern

Deep Learning Intern

Research Engineer Intern

Research Engineer Intern

AI Infra Onboard Performance Intern

AI Infra Onboard Performance Intern