PlusAI · 1 week ago
Senior Machine Learning Infrastructure Engineer
Santa Clara, CA
Full-time
Hybrid
Mid, Senior Level
$160K/yr - $200K/yr
3+ years expPlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. As a Senior ML Infrastructure Engineer, you will design scalable architectures for handling large data sets and build robust pipelines for model versioning and experiment tracking, while managing large-scale GPU clusters.
Artificial Intelligence (AI)Autonomous VehiclesLogisticsTransportation
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
Qualification
Required
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
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
Benefits
Catered free lunch
Unlimited snacks and beverages
Highly competitive salary and benefits package, including 401(k) plan
Company
PlusAI
PlusAI is an artificial intelligence company pioneering AI-based virtual driver software for factory-built autonomous trucks.
201-500 employees
H1B Sponsorship
PlusAI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2023 (1)
Funding
Current Stage
Growth StageTotal Funding
$520.03MKey Investors
HSGGSR Ventures
2021-03-01Series C· $100.15M
2020-11-23Series B· $286.18M
2018-11-15Series A· $32M
Leadership Team
David Liu
Founder & CEO
Hao Zheng
Co-Founder & CTO
Recent News
2026-01-12
2026-01-09
Company data provided by crunchbase