Software Engineer, ML Infra
$180k - $225kNewsBreak
Machine Learning Infrastructure Engineer
We're hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the models behind our Ads and Recommendations products. You'll join a small, high-ownership team that ships platform improvements end-to-end—partnering with product and data teams, reducing latency and cost, and shortening the path from an idea to a safely launched model.
You'll work across the ML lifecycle: making training faster and more reliable, improving model serving performance, and strengthening our feature/embedding platform so models stay fresh and consistent between offline and online use. We're looking for someone who can take real ownership, finish what's started, and raise the bar on stability and developer experience.
If you like building reliable systems that make ML teams move faster—and you enjoy turning complexity into simple, durable solutions—we'd love to talk.
Responsibilities
- Design and develop machine learning infrastructure.
- Own and enhance core components of the ML infrastructure, including systems for offline and online model training, model pipeline health monitoring, model serving, feature authoring, and feature serving.
- Proactively address ML infrastructure issues that may impact production.
- Collaborate with ML engineers to build robust model pipelines utilizing the ML infrastructure.
Requirements
- Education: Bachelor's degree in a related field with 5+ years of relevant experience, or MS/PhD in a related field with 3+ years of relevant experience
- Experience delivering production systems that support ML use cases (training or serving).
- Proven track record in building and maintaining large-scale distributed backend systems
- Proficient in Python, with a strong understanding of object-oriented languages such as C++ or Java
- Strong problem solving skills with good teamwork and communication skills
Preferred Qualifications
- Experience leading cross-team projects from design through stable rollout.
- Experience iterating on ML models and shipping them to production.
Benefits
We offer competitive benefits package:
- Health, dental, and vision care for you and your family
- Top-tier 401(K) plan with company matching
- Paid time off and paid holidays
- Paid parental leave
- FSA and commuter benefits programs
- Team activity budget
The US base salary range for this full-time position is listed below. Pay may vary based on a number of factors including job-related skills, level, experience, geographic location and relevant education or training. At NewsBreak, we design our overall rewards package to attract top talents. Depending on the position, the role may also be eligible for discretionary bonus and options. Your recruiter can share more details during the hiring process.
Annual Base Pay Range
$180,000 - $225,000 USD
$180k - $225k
...visit About the role:We’re hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the... ...from an idea to a safely launched model.You’ll work across the ML lifecycle: making training faster and more reliable, improving...SuggestedFull timeLocal areaWork from home$147k - $210k
...computing concepts, consensus algorithms (e.g., Paxos).Google's software engineers develop the next-generation technologies that change how... ...performant bulk data loading pathways(essential for emerging AI/ML workloads), introducing advanced mutation semantics and data life...Suggested$272k - $431.25k
We are seeking a Principal AI and ML Infra Software Engineer, GPU Clusters at NVIDIA to join our Hardware Infrastructure team. As an Engineer, you will have a pivotal role in enhancing efficiency for our researchers by implementing progressions throughout the entire stack...SuggestedFull time$207k - $300k
...Dev and AI Infra Software Engineer, Release and Test Infra, XR Share Dev and AI Infra Software Engineer, Release and Test Infra, XR corporate... ..., development, and architecture. ~ Experience applying AI/ML techniques to software engineering problems. Preferred...Suggested$165.2k - $223.6k
...responsible for delivering high quality and fresh ML model training data, and providing... ...station.Key job responsibilitiesAs an ML Engineer you will:Lead development of services and... ...3+ years of non-internship professional software development experience- 2+ years of non-...SuggestedInternshipLocal areaWorldwideFlexible hours$275.8k - $340.5k
...self-driving vehicles, join us. About the team: The AV ML Infra team at GM builds ML infrastructure designed to meet the unique... ...efficient ML experimentation, enhance the productivity of ML engineers, and drive the adoption of cutting-edge ML techniques. Our...Full timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$147k - $210k
...Implement solutions in one or more specialized ML areas, utilize ML infrastructure, and... ...accessible technologies.Google's software engineers develop the next-generation technologies... ...businesses at Google. The YouTube Ads ML Infra team empowers YouTube Ads ML advancement...$193.93k - $291.15k
...team of high-output generalists where ML and systems engineering converge. This is not a "run the models... ...eval autonomy path, across ML and infra boundaries.About YouBS in Computer Science... ...area.4+ years of industry software engineering experience with Python fluency...Immediate startFlexible hours$193.3k - $261.5k
...Senior Software Development Engineer, AWS Infra Supply Chain Automation Job ID: 10489640 | Amazon.com Services LLC AWS Infrastructure Services owns the design, planning, delivery, and operation of all AWS global infrastructure. In other words, we’re the people who...InternshipLocal areaFlexible hours- ...role and team We are looking for senior software engineers to build the backbone of Neural... ...generation, and log augmentation, along with the ML pipelines used to train and validate... ...distributed services Collaborate closely with Infra, Autonomy, Research and other product...Full timeFor contractorsFor subcontractorCasual workWork at officeRemote workDay shift
$180k - $210k
...work across distributed systems, production observability, and ML infra integrations to deliver an experience that feels instant,... ...most impactful roadmap investments. You are: ~3+ years of software engineering experience building production distributed systems or cloud-...Work at officeHome officeFlexible hours$209k - $313k
...and other digital services.Snap Engineering teams build fun and... ...critical role in scaling our ML Infrastructure, optimizing AI... ...impactful.We’re looking for a Software Engineer to join the ML Platform... ...deprecation. With the significant infra cost of this area, our team also...Live inWork at officeLocal area$206.5k - $258.1k
...future generations. Role Summary Rivian Autonomy is building an ML Infrastructure team to give hundreds of ML engineers a training platform they can trust at fleet scale. We are seeking a Staff Software Engineer to help design and build the platform that trains and...Full timeContract workTemporary workPart timeLocal areaShift work$70 per hour
...and tens of billions in simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's fully autonomous driving technology... ...us build the data platform and infrastructure to support ML launch evals for the Waymo Driver. Waymo interns partner with...Hourly payFull timeInternshipSummer internshipRelocation package$174k - $252k
...visualization workflows.Support new and exciting ML paradigms (such as horizontal scaling for... ...experience.5 years of experience with software development in one or more programming... ...open-source projects.Google's software engineers develop the next-generation technologies...$174k - $252k
...Senior Software Engineer, Performance, SOC Architect Share Senior Software Engineer, Performance, SOC Architect corporate_fare Google place... ...and a passion for developer productivity. ~ Experience with ML frameworks such as TensorFlow, JAX, and PyTorch, or ML...$164k - $282k
...workflow service at scale.We are looking for a visionary Staff Software Engineer to lead the architectural design and implementation of our... ...always on for our customers.AI-Driven Automation: Integrate AI/ML models to automate system functionality, including predictive...Temporary work$120k - $195k
...’s AI model training, feature engineering and serving with hundreds of billions... ...scale feature engineering infra for all AI use cases from... ...frameworks, data infra, compute software, and hardware to harness the power... ...with hundreds of new ML models per quarter using thousands...For contractorsWork experience placementWork at officeFlexible hours$117k - $234k
...Role Summary:We are seeking a talented and passionate Senior, Software Engineer (Backend) for our team. You will be part of Catalog Spec Engineering... ..., and problem solving.Experience working with generated AI/ML technologies.Experience with NoSQL databases like Cosmos DB,...Full timeTemporary workPart timeWork at office- ...page, in any of our markets, a decision engine works behind the scenes to choose the right... ...Airflow) that feed our algorithms and AI/ML models. Partner with data scientists... ...ll Bring: ~5+ years building production software, including 3+ years in Java and/or Python...Full timeRemote work
- ...Full Stack Engineer – Digital Payments About Gen: Gen is a global company dedicated to powering Digital Freedom through its trusted... ...build intelligent integrations is a strong plus. Exposure to AI/ML concepts and the ability to evaluate and integrate AI‑driven...Work experience placementFlexible hours
$125k - $175k
...hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains,... ...launched model. You'll work across the ML lifecycle: making training faster and... ...related discipline. Familiarity with software development processes, including version...Full timeWork experience placementLocal area- ...you, we'd love to talk. Role Overview: As a Full Stack Software Engineer, you will be a pivotal force in developing, deploying, and... ...expertise in designing, scaling, and optimizing enterprise-grade ML or data-intensive systems. Preferred Qualifications:...
$85.4k - $192.9k
...designing, developing, testing and maintaining software applications and products to meet... ...customer requirements. Full-stack software engineering roles, who can develop all components of... ...MS SQL Server, PostgreSQL, and MySQL. ~ ML Training: Google Vertex AI (Studio, Training...Immediate startFlexible hours$150k
...-class researchers, data scientists, and engineers, tackling the most fundamental and impactful... ...to collaborate with Researchers and ML Engineers to produce AI applications and... ...functional team that's responsible for the full software development life cycle, from conception...$147k - $210k
...Implement solutions in one or more specialized ML areas, utilize ML infrastructure, and... ...:Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science,... ...launching features.2 years of experience with software development in one or more programming languages...Local area- ...system-level results into clear feedback for engineering and leadership, and help accelerate... ...analyses to introspect autonomous driving software performance at subsystem interfaces across... ...Propose and develop new statistical and ML methods to quantify performance and identify...Full timeLocal areaWork from home
$130k - $260k
...shaping the long-term technical vision, architecture, and engineering excellence of our enterprise-scale AI/ML systems and platforms. You will work across teams to... ...building and operating large-scale, distributed software systems. Proven track record as a senior individual...Full timeContract workTemporary workPart time$162k - $260k
...efficient and accessible for all. We’re searching for a Senior Software Engineer, Perception to join our autonomous driving team. In this role,... ...testing to produce robust, well-tested systems.Scale ML Engineering Pipelines: Build and scale robust ML pipelines to...Work at officeLocal area3 days per week$174k - $252k
...minimal latency.Collaborate with cross-functional product and engineering teams to integrate features—like seamless cross-platform video... ...working with algorithms, data structures, complexity analysis and software design.Experience with Android system-level development or...Worldwide
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Software Engineer, ML Infra. Be the first to apply!
- software engineer amazon Mountain View, CA
- information technology software engineer Mountain View, CA
- federal - software developer Mountain View, CA
- graduate software developer Mountain View, CA
- rust software engineer Mountain View, CA
- software development engineer (robotics engineer) Mountain View, CA
- software engineer remote Mountain View, CA
- junior software developer internship Mountain View, CA
- remote entry level software developer Mountain View, CA
- software engineer contract Mountain View, CA




