Machine Learning Infrastructure Engineer Intern

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.


Ready to get hands-on with real-world, large-scale data challenges? We’re seeking a Machine Learning Infrastructure Engineer Intern to join us in a project that focuses on identifying the bottlenecks and implementing high-performance custom kernels (using CUDA, Triton, or C++) to accelerate BEV model training. Uniquely, this internship will also explore the utilization of LLMs (Large Language Models) to assist in high-performance code generation, kernel optimization, and automated performance profiling with Nsight and Pytorch profiler