Senior Machine Learning Infrastructure Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Plus · 5 hours ago

Senior Machine Learning Infrastructure Engineer

Plus 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 managing large datasets and build robust pipelines for model versioning and experiment tracking, while collaborating with cross-functional teams to optimize performance and reliability of the ML platform.

Artificial Intelligence (AI)Autonomous VehiclesLogisticsTransportation
check
H1B Sponsor Likelynote

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

Machine Learning InfrastructureDistributed SystemsDeep Learning FrameworksContainerization TechnologiesPythonC++SQLCloud PlatformsSoftware Design SkillsCommunication SkillsAdaptability

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

Plus

twittertwitter
company-logo
Plus is an AI company whose mission is to build “driving intelligence” to power an autonomous future

H1B Sponsorship

Plus 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 Stage
Total Funding
$520.03M
Key Investors
HSGGSR Ventures
2021-03-01Series C· $100.15M
2020-11-23Series B· $286.18M
2018-11-15Series A· $32M

Leadership Team

leader-logo
David Liu
Founder & CEO
linkedin
leader-logo
Hao Zheng
Co-Founder & CTO
linkedin
Company data provided by crunchbase