The Role
As a Senior ML Infrastructure Engineer, you will design scalable architectures for ML, build robust data pipelines, manage GPU clusters, and drive improvements in CI/CD workflows while collaborating with cross-functional teams.
Summary Generated by Built In
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.
As a Senior ML Infrastructure Engineer at Plus, you will design scalable architectures capable of handling petabytes of data while ensuring optimal performance for both training and inference phases. You will build robust pipelines for managing model versioning systems and experiment tracking frameworks, which are essential for maintaining reproducibility across experiments. Additionally, you will be responsible for managing large-scale GPU clusters. This role offers unparalleled opportunities—both technically and professionally—for individuals passionate about solving challenging problems using modern cloud-native technologies. Ideal candidates thrive in environments that leverage tools such as Docker containers orchestrated via Kubernetes clusters, seamlessly integrated with state-of-the-art deep learning frameworks like PyTorch or TensorFlow. If you are eager to push the boundaries of what's possible in machine learning infrastructure and contribute to cutting-edge solutions, this position is an excellent fit!
Responsibilities:
- Design and develop scalable, high-performance systems for training, inference, deploying, and monitoring ML models at scale.
- Build and maintain efficient data pipelines, model versioning systems, and experiment tracking frameworks.
- Collaborate with cross-functional teams, including ML researchers and engineers, to identify bottlenecks and improve platform usability.
- Implement distributed systems and storage solutions optimized for machine learning workloadsDrive improvements in CI/CD workflows for ML models and infrastructure.
- Ensure high availability and reliability of the ML platform by implementing robust monitoring, logging, and alerting systems.
- Stay current with industry trends and integrate relevant tools and frameworks to enhance the platform.
- Mentor junior engineers and contribute to a culture of technical excellence
- Ensure that your work is performed in accordance with the company’s Quality Management System (QMS) requirements and contribute to continuous improvement efforts.
- Ensure team compliance with QMS, monitor quality, and drive process improvements.
Required Skills:
- Phd or MS in Computer Science, Electrical Engineering, or related field
- Good oral and written communication skills
- Phd new grad or Masters with 3+ years of software engineering experience with a focus on ML infrastructure or distributed systems.
- Proficiency in in Python, C++, SQL
- Deep understanding of containerization, orchestration technologies, distributed ML workload, and experiment tracking tools (e.g., Docker, Kubernetes, multiprocessing, Kubeflow, and mlflow)
- Deploy and manage resources across multiple cloud platforms (AWS, GCP, or on-prem environments)
- Proficiency in at least one deep learning framework, such as PyTorch and data pipeline tools (e.g., Apache Airflow, Prefect).
- Strong knowledge of distributed systems, databases, and storage solutions.
- Extensive software design and development skills.
- Ability to learn and adapt to new technologies and contribute in a productive environment.
Preferred Skills:
- Familiarity with fundamental deep learning architectures, such as Convolutional Neural Networks (CNNs) and Transformer models
- Experience in building large-scale ML datasets, MLOps pipelines, and distributed computing frameworks like Ray
- Experience working with autonomous vehicles or robotics
Salary Range:
- $160,000 - $200,000 a year
Our compensations (cash and equity) are determined based on the position, your location, qualifications, and experience.
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.
Skills Required
- PhD or MS in Computer Science, Electrical Engineering, or related field
- 3+ years of software engineering experience with a focus on ML infrastructure or distributed systems
- Proficiency in Python, C++, SQL
- Deep understanding of containerization and orchestration technologies
- Proficiency in at least one deep learning framework, such as PyTorch
Plus Compensation & Benefits Highlights
The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Plus and has not been reviewed or approved by Plus.
-
Leave & Time Off Breadth — Unlimited PTO in addition to company holidays and flexible work arrangements are offered, indicating broad time-off flexibility. This setup signals strong support for taking time away from work.
-
Healthcare Strength — Tiered medical, dental, and vision options allow employees to select coverage that fits their needs. This breadth of core health coverage aligns with a comprehensive benefits approach.
-
Wellbeing & Lifestyle Benefits — Daily catered lunches at key offices and company-sponsored professional development add meaningful day-to-day and growth-oriented perks. These offerings enhance overall wellbeing and workplace experience.
Plus Insights
Am I A Good Fit?
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.
Success! Refresh the page to see how your skills align with this role.
The Company
What We Do
Plus is a global provider of highly automated driving and fully autonomous driving solutions. Named by Forbes as one of America's Best Startup Employers and Fast Company as one of the World’s Most Innovative Companies, Plus's customers are already operating its product on the road today. Working with one of the largest companies in the U.S., vehicle manufacturers and others, Plus is making transportation safer and greener. Plus has received a number of industry awards and distinctions for its transformative technology and business momentum from Fast Company, Insider, Consumer Electronics Show, AUVSI, and others. For more information, visit www.plus.ai
