Machine Learning Infrastructure Engineer Jobs

The Personalization team makes deciding what to play next easier and more enjoyable for every listener. From Blend to Discover Weekly, we're behind some of Spotify's most-loved features. We built them by understanding the world of music and podcasts better than anyone else. Join us and you'll keep millions of users listening by making great recommendations to each and every one of them. Surfaces Moments is a ML team within the Personalization Mission focused on creating moment-based experiences across Spotify surfaces. The team owns and evolves the experiences that help listeners quickly connect with the content that matters most to them, including the Home Shortcuts experience and the underlying intelligence that powers it. By combining cutting-edge machine learning, recommendation systems, and product thinking, the team delivers highly relevant, personalized experiences to millions of listeners around the world. As a Staff Machine Learning Engineer, you will help shape the future of personalized discovery and engagement at Spotify. You'll work at the intersection of recommendation systems, large language models, and production-scale machine learning infrastructure to build experiences that delight users and drive meaningful impact. This role is ideal for someone who enjoys taking models from research to production, driving technical direction in ambiguous problem spaces, and solving complex personalization challenges at global scale. What You'll Do <li style="font-size: 11pt; font-family: Arial, san

We believe that the way people interact with their finances will drastically improve in the next few years. We’re dedicated to empowering this transformation by building the tools and experiences that thousands of developers use to create their own products. Plaid powers the tools millions of people rely on to live a healthier financial life. We work with thousands of companies like Venmo, SoFi, several of the Fortune 500, and many of the largest banks to make it easy for people to connect their financial accounts to the apps and services they want to use. Plaid’s network covers 12,000 financial institutions across the US, Canada, UK and Europe. Founded in 2013, the company is headquartered in San Francisco with offices in New York, Seattle, Washington D.C., Raleigh, London, and Amsterdam. We are the Data Foundation & AI team within Plaid’s Data organization. Our mission is to build the shared ML and AI infrastructure that powers intelligent capabilities across Plaid’s product suite. We develop the foundational systems, models, and data assets that transform Plaid’s unique financial network data into scalable, general-purpose representations that teams across the company can leverage. Our work spans the full ML lifecycle — from large-scale data curation and model pretraining to production serving, evaluation, and monitoring. As part of the team, you’ll work at the intersection of machine learning infrastructure, applied AI, and distributed systems, helping establish the core AI platform that enables innovation across Plaid. As a Staff Machine Learning Engineer, you will lead the technical strategy and development of Plaid’s foundation models, driving key decisions across pretraining objectives, model architecture, and fine-tuning approaches that power a wide range of downstream product applications. You will serve as the technical lead for the full machine learning lifecycle, overseeing everything from data curation and experimentation to production deployment,

Reddit is a community of communities. It’s built on shared interests, passion, and trust, and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information. For more information, visit www.redditinc.com . Who We Are: The Machine Learning Platform team at Reddit is a high-impact team that owns the infrastructure that powers recommendations, content discovery, user and content quantification, while directly impacting other teams such as Growth, Ads, Feeds, and Core Machine Learning teams. What You’ll Do: As a Staff ML Infrastructure Engineer, you will lead development of a platform for large scale ML models at Reddit. Design end-to-end model lifecycle patterns (MLOps) to boost velocity of development for ML engineers, including data preparation, model management, experiment tracking, and more Zero-to-one development and support of a graph ML codebase and platform that abstracts away common patterns and enables greater model scalability and iteration Collaborate with ML engineers on performance tuning, including improving model training time, efficiency, and GPU training costs in a large, distributed ML training environment Optimize batch data processing within a data warehouse and with tools such as Apache Beam, Apache Spark, Ray Data, and more Architect pipelines to build and maintain massive graph data structures on the order of billions of nodes and tens of billions of edges Who You Might Be: 8+ years of experience in ML infrastructure, including model training and model deployments Hands-on experience with ML optimization, including memory and GPU profiling Deep experience with cloud-based technologies for supporting an ML platform, including tools like GCP BigQuery, Google Cloud Storage, infrastructure-as-code (

Reddit is a community of communities. It’s built on shared interests, passion, and trust, and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information. For more information, visit www.redditinc.com . Who We Are: The Machine Learning Platform team at Reddit is a high-impact team that owns the infrastructure that powers recommendations, content discovery, user and content quantification, while directly impacting other teams such as Growth, Ads, Feeds, and Core Machine Learning teams. What You’ll Do: As a Senior ML Infrastructure Engineer, you will lead development of a platform for large scale ML models at Reddit. Design end-to-end model lifecycle patterns (MLOps) to boost velocity of development for ML engineers, including data preparation, model management, experiment tracking, and more Zero-to-one development and support of a graph ML codebase and platform that abstracts away common patterns and enables greater model scalability and iteration Collaborate with ML engineers on performance tuning, including improving model training time, efficiency, and GPU training costs in a large, distributed ML training environment Optimize batch data processing within a data warehouse and with tools such as Apache Beam, Apache Spark, Ray Data, and more Architect pipelines to build and maintain massive graph data structures on the order of billions of nodes and tens of billions of edges Who You Might Be: 5+ years of experience in ML infrastructure, including model training and model deployments Hands-on experience with ML optimization, including memory and GPU profiling Deep experience with cloud-based technologies for supporting an ML platform, including tools like GCP BigQuery, Google Cloud Storage, infrastructure-as-code

Our Perception team is responsible for using our sensor data to understand the complex and dynamic environments where we drive. In this role, you will have access to the best sensor data in the world and an incredible infrastructure for testing and validating your algorithms. We are creating new algorithms for segmentation, tracking, classification, and high-level scene understanding, and you could work on any (or all!) of these components. As a Senior Staff ML Engineer, you will lead the development of machine learning algorithms that can range in influence from onboard autonomy to offboard autonomy and validation. You will collaborate closely with other teams specializing in Prediction, Planning, Simulation, and Safety Validation, influencing our overall technical stack. Your role will look at problems in a way that crosses team boundaries to prototype new approaches that influence the long term technical direction of multiple organizations within the company. The impact of the role can be in the form of impacting immediate company milestones to leading forward-looking exploratory projects. Responsibilities Develop new algorithms to understand the scene around the robot, and how that scene would evolve through time Build multi-modal foundation models for on-vehicle and offline applications Develop new algorithms to apply generative AI to simulation to improve the realism of our offline validation systems Leverage our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field Provide technical mentorship to the broader group of ML developers at Zoox Collaborate with engineers on Prediction, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments Qualifications BS, MS, or PhD degree in computer science or related field Experience with training and deploying Deep Learning models on sensor data-Experience

Figma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma’s platform helps teams bring ideas to life—whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI. From idea to product, Figma empowers teams to streamline workflows, move faster, and work together in real time from anywhere in the world. If you're excited to shape the future of design and collaboration, join us! Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma. You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure. This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities. This is a full time role that can be held from one of our US hubs or remotely in the United States. What you’ll do at Figma: Design, build, and productionize ML models for Search, Discovery, Ranking, Retrieval-Augmented Generation (RAG), and generative AI features. Build and maintain scalable data pipelines to collect high-quality training and evaluation datasets, including annotation systems and human-in-the-loop workflows. Collaborate with AI researchers to iterate on datasets, evaluation metrics, and model architectures to improve quality and relevance. Work with product engineers to define and deliver impactful AI features across Figma’s platform. Partner with infrastructure engineers to develop and optimize systems for training, inference, monitoring, and deployment. Explo

The Prediction & Behavior ML team is responsible for developing machine learning (ML) algorithms that learn and predict behaviors from data, applying them both on-vehicle to influence driving behavior and off-vehicle to provide ML capabilities to simulation and validation. Given the tight integration of behavior forecasting and motion planning, our team collaborates closely with the Planner team to advance overall vehicle behavior. We also work closely with our Perception, Simulation, and Systems Engineering teams to accelerate our ability to validate our driving performance. As a Learned Trajectory Machine Learning Engineer you will be responsible for developing deep learned models that produce trajectories for our vehicles to drive. Given the tight integration of behavior prediction and motion planning, you will closely collaborate with the Planner and Perception teams in the advancement of our overall vehicle behavior. In this role, you will: You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for our autonomous vehicle. You will also work on techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort etc. You will leverage our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field You will develop metrics and tools to analyze errors and understand improvements of our systems You will collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments Qualifications BS, MS, or PhD degree in computer science or related field Experience with training and deploying transformer-based model architectures and reinforcement learning Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines Fluency in C++

The Prediction & Behavior ML team is responsible for developing machine learning (ML) algorithms that learn and predict behaviors from data, applying them both on-vehicle to influence driving behavior and off-vehicle to provide ML capabilities to simulation and validation. Given the tight integration of behavior forecasting and motion planning, our team collaborates closely with the Planner team to advance overall vehicle behavior. We also work closely with our Perception, Simulation, and Systems Engineering teams to accelerate our ability to validate our driving performance. As a Learned Trajectory Machine Learning Engineer you will be responsible for developing deep learned models that produce trajectories for our vehicles to drive. Given the tight integration of behavior prediction and motion planning, you will closely collaborate with the Planner and Perception teams in the advancement of our overall vehicle behavior. In this role, you will: You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for our autonomous vehicle. You will also work on techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort etc. You will leverage our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field You will develop metrics and tools to analyze errors and understand improvements of our systems You will collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments Qualifications BS, MS, or PhD degree in computer science or related field Experience with training and deploying transformer-based model architectures and reinforcement learning Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines Fluency in Pyt

About the Team DoorDash is building the world's most reliable on-demand, logistics engine for delivery. We are continuing to grow rapidly and expanding our Engineering offices globally! We are looking for Software Engineers, Machine Learning to build and maintain a large scale 24x7 global infrastructure system that powers DoorDash's 3-sided marketplace of Consumers, Merchants and Dashers. About the Role As a Software Engineer, Machine Learning, you’ll be conceptualizing, designing, implementing, and validating algorithmic improvements to the catalog system and our product knowledge graph at the heart of our fast-growing grocery and retail delivery business. You will use our robust data and machine learning infrastructure to implement new ML solutions to make our product knowledge graph accurate, standardized, semantically rich, easily discoverable, and extensible. We’re looking for someone with a command of production-level machine learning and experience with solving end-user problems who enjoys collaborating with multi-disciplinary teams. This role is hybrid with some in-office time expected and will report to an Engineering Manager. You're excited about this opportunity because you will... Develop advanced machine learning models to improve ads efficiency and quality. Design and build optimization algorithms for budget pacing and automated bidding to achieve various advertising goals. Establish a data-driven framework to understand how the bid density and market competitiveness would affect advertising value and platform revenue. Develop new data solutions (eg. embeddings and consumer profiles) to target the relevant audience. Be responsible for the end-to-end ML lifecycle, including ideation, offline model training, online shadowing/deployment, experimentation, and post-launch monitoring/measurement. Build and extend the current data/ML infrastructure to empower Ads data applications including data analysis, ML modeling, and experimentati

About the Role As a Senior Software Engineer, Machine Learning, you’ll be conceptualizing, designing, implementing, and validating algorithmic improvements to the catalog system and our product knowledge graph at the heart of our fast-growing grocery and retail delivery business. You will use our robust data and machine learning infrastructure to implement new ML solutions to make our product knowledge graph accurate, standardized, semantically rich, easily discoverable, and extensible. We’re looking for someone with a command of production-level machine learning and experience with solving end-user problems who enjoys collaborating with multi-disciplinary teams. We expect Senior Engineers to be able to own projects end-to-end and mentor their junior and mid level peers. This role is hybrid with some in-office time expected and will report to an Engineering Manager. You're excited about this opportunity because you will... Develop advanced machine learning models to improve core parts of your teams scope. Be responsible for the end-to-end ML lifecycle, including ideation, offline model training, online shadowing/deployment, experimentation, and post-launch monitoring/measurement. Work on creating novel ML & AI solutions that push both our business and the industry forward. Scale our systems and services to fuel the growth of our business. Completely disrupt logistics by tackling bleeding-edge, technical problems We're excited about you because you have... M.S., or PhD. in a technical field such as computer science, mathematics, statistics, physics or equivalent. 5+ years ML industry experience with a solid understanding of machine learning algorithms and fundamentals. Proficiency in using AI coding tools (e.g., Claude Code, Codex, Cursor) in the full software development lifecycle, including designing, generating code, testing, monitoring and releasing software Experience building data/feature engineering pipelines at scale using Pyspark and Snowflak

Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems. We’re training and deploying frontier models for enterprises who are building AI systems. We believe that our work is instrumental to the widespread adoption of AI and we are looking for folks that want to be part of that. We obsess over what we build. Each one of us is responsible for contributing to increasing the capabilities of our models and the value they drive for our customers. Cohere is a team of researchers, engineers, designers, and more, who are all passionate about their craft. We are a global technology company headquartered in Toronto with key offices in London, New York City, San Francisco, Montreal, Paris, Berlin and Seoul. Join us! Why this role? As Cohere continues to grow, the size and complexity of our programs has continued increasing over time. To help us manage this, we are looking to bring in exceptional Technical Program Managers (TPM) to manage this! At Cohere, TPMs are pivotal to our success, being seen as operational experts specializing in cross-functional efforts. Great TPMs at Cohere are well-rounded, executing with precision at the tactical level, while partnering with senior leaders and shaping the big picture at a strategic level, and the opportunities are endless in growing the scale and impact of their work. Cohere’s TPMs are often described by their stakeholders as organized, execution-oriented, pragmatic, and adaptable to the needs of the company and the programs they lead. As a reward for the depth and breadth of expertize that you bring to the table, you will get to work alongside some of the most talented engineers in the world on building cutting edge AI technology. No day will be the same as the one before, and the dynamic high growth environment will be absolutely perfect for anyone excited about solving novel problems, bringing

About the Team The Monetization team is a new cross-functional group working across engineering, product, research, and design to build the foundational systems that will help OpenAI scale access to intelligence responsibly. Our mission is to develop user-first, privacy-preserving monetization products—including next-generation ads experiences—that strengthen user trust, unlock economic opportunity, and support OpenAI’s long-term innovation. Monetization plays a critical role in enabling OpenAI to continue pushing the boundaries of AI capabilities while ensuring the benefits of AGI are broadly shared. We believe monetization must be aligned with user value, uphold rigorous privacy and safety standards, and sustain a healthy ecosystem of developers and businesses. This team operates in a greenfield environment and moves quickly through prototyping, experimentation, and iterative deployment. We partner closely with Product, Design, and Research to bring research breakthroughs into real-world systems at global scale. About the Role We’re looking for an experienced Software Engineer to help build the machine learning infrastructure that powers OpenAI’s monetization and ads systems. In this foundational role, you’ll design and develop the platform layer that enables teams to build, train, deploy, serve, monitor, and continuously improve machine learning models used across advertising and monetization products. You’ll work across the full ML lifecycle, from large-scale data pipelines and feature infrastructure to training systems, model serving, experimentation platforms, and monitoring frameworks. The systems you build will support high-throughput, low-latency advertising workloads while maintaining strict standards for reliability, privacy, security, and performance. This role sits at the intersection of machine learning systems, distributed infrastructure, and monetization, offering the opportunity to shape the core platforms that help translate model innovation into m

Who We Are At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves. Across our multiple offices globally, we are united by our core principles: We Before Me , Do the Right Thing , and Get Things Done . These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er. OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more. About the Opportunity We are seeking an AI Infrastructure Engineer to design and build the foundational systems that power next-generation AI agents. This role sits at the intersection of research and production engineering, focusing on scalable long-term memory , knowledge graph infrastructure , and robust context management systems . You will transform cutting-edge research concepts (e.g., Context Graphs, GraphRAG, agent memory architectures) into auditable, maintainable, and production-grade infrastructure that enables reliable, stateful AI systems. What You’ll Be Doing Architect and implement long-term memory infrastructure for AI agents , including auditable, versioned, and rollback-capable middleware that supports persistent, stateful reasoning. Design and own end-to-end GraphRAG and knowledge graph pipelines , from ingestion and schema design to graph indexing, retrieval optimization, and integration with LLM-based reasoning systems. Build scalable context management infrastructure , elevating context from prompt-level techniques to structured, maintai

OKX will be prioritising applicants who have a current right to work in Singapore, and do not require OKX's sponsorship of a visa. Who We Are At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves. Across our multiple offices globally, we are united by our core principles: We Before Me , Do the Right Thing , and Get Things Done . These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er. OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more. About the Opportunity We are seeking an AI Infrastructure Engineer to design and build the foundational systems that power next-generation AI agents. This role sits at the intersection of research and production engineering, focusing on scalable long-term memory , knowledge graph infrastructure , and robust context management systems . You will transform cutting-edge research concepts (e.g., Context Graphs, GraphRAG, agent memory architectures) into auditable, maintainable, and production-grade infrastructure that enables reliable, stateful AI systems. What You’ll Be Doing Architect and implement long-term memory infrastructure for AI agents , including auditable, versioned, and rollback-capable middleware that supports persistent, stateful reasoning. Design and own end-to-end GraphRAG and knowledge graph pipelines , from ingestion and schema design to graph indexing, retrieval optimization, and integration with LLM-based reasoning systems. Build scalable context management inf

About the Role: The Machine Learning team at Tubi drives the innovation behind personalized user experiences. With the largest inventory in the industry and hundreds of millions of viewers, we tackle problems in the space of recommendations, search, content understanding, and ads optimization that shape the future of streaming. We are seeking a Director of Machine Learning Engineering and Infrastructure to lead a hybrid team bridging advanced ML engineering with world-class infrastructure design. In this role, you will own the strategic direction and execution for scaling our machine learning capabilities while ensuring our distributed systems and infrastructure can support innovation at massive scale. You will combine technical depth with leadership excellence to guide teams that deliver both foundational ML systems and high-performance distributed services. This is a hybrid role for our Toronto office. What You'll Do: Lead and manage high-performing teams across ML engineering and ML infrastructure, fostering a culture of innovation, collaboration, and growth. Define and execute the strategic roadmap for ML systems, including recommendation, personalization, and ads optimization. Oversee the design, development, and deployment of scalable ML pipelines: data ingestion, feature engineering, model training, evaluation, and serving. Architect distributed systems to support ML workloads at scale, ensuring reliability, observability, and operational excellence. Partner closely with Product, Engineering, and Content teams to align on business goals and deliver impactful ML-driven experiences. Support best practices in experimentation, evaluation, and ML system monitoring. Ensure cost efficiency, scalability, and performance in ML infrastructure investments. Your Background: 10+ years of industry experience spanning machine learning engineering and distributed systems. 3+ years of leadership and management experience, with a proven ability to build and lead strong t