Source Job

US

  • Design, train, and improve large-scale machine learning models for recommendation and personalization, leveraging modern deep learning architectures.
  • Own end-to-end delivery of major ML system components, from problem framing to production rollout, with cross-functional partners.
  • Optimize distributed training, model efficiency, and online inference performance while ensuring low-latency, high-throughput production systems.

Python PyTorch Deep Learning Distributed Training ML Infrastructure

20 jobs similar to Senior Machine Learning Infrastructure Engineer, Embedding Platform

Jobs ranked by similarity.

US

  • Architect and lead the development of next-generation, large-scale machine learning techniques.
  • Define and execute the ML strategy, identifying opportunities to enhance personalization and recommendation quality across Reddit.
  • Partner with ML infrastructure teams to build high-performance, distributed training systems that efficiently scale across multiple GPUs and cloud environments.

Reddit is a community of communities, built on shared interests, passion, and trust. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information.

US

  • Independently own high-value optimization initiatives across training, inference, or launch-readiness for important Ads ML workloads.
  • Diagnose bottlenecks in real production systems using profiling, benchmarking, and observability.
  • Build performance tooling, optimization playbooks, and efficiency primitives that benefit multiple teams.

Reddit is a community of communities built on shared interests and authentic conversations. With 100,000+ active communities and approximately 126 million daily active unique visitors, Reddit has a flexible workforce and values collaboration.

US

  • Drive GenAI-powered recommendations leveraging Reddit's unique community landscape and conversational data.
  • Set technical direction for relevance and feeds teams, influencing strategy across multiple orgs.
  • Mentor and grow senior and staff engineers while fostering a strong engineering culture.

Reddit is a community of communities built on shared interests, passion, and trust, hosting the most open conversations. With 100,000+ communities and 130 million daily active users, it is one of the internet's largest information sources.

Australia

  • Design, implement, and refine machine learning models to deliver personalized recommendations.
  • Lead the design of online and offline experiments to validate model performance and make data-driven decisions.
  • Work closely with product and engineering teams to deploy new recommendation features and drive alignment across teams.

Canva is redefining how the world experiences design by providing a platform for creating stunning visual content. With over 200 million monthly users and a culture of flexibility and innovation, the team is committed to empowering creativity and collaboration.

North America 5w PTO

  • Own and improve the machine learning models and systems powering the Home feed and Shortcuts experience.
  • Design and build personalized recommendations serving millions of Spotify listeners globally.
  • Train, fine-tune, and optimize large language models using techniques like SFT and distillation.

Spotify is a leading audio streaming platform that delivers personalized music and podcast experiences to millions of listeners globally. The Personalization team is a large, collaborative group behind iconic features like Discover Weekly, with a culture of innovation and user-centricity.

Canada

  • Lead a team of platform engineers to build and operate ML training and serving infrastructure, including GPU and low-latency serving.
  • Combine strong people leadership with technical judgment in ML infrastructure, partnering with senior ICs and cross-functional teams.
  • Drive delivery, operational health, and evaluate modern ML tooling to support company-wide ML priorities.

Affirm is reinventing credit to make it more honest and friendly, offering buy now pay later solutions without hidden fees. It is a remote-first company with a strong engineering culture, prioritizing people and providing competitive benefits.

$206,261–$330,017/yr
US

  • Design, build, deploy, and optimize machine learning models that process large volumes of complex, unstructured data.
  • Develop and maintain scalable ML pipelines capable of supporting millions of documents and diverse customer requirements.
  • Lead technical initiatives from early experimentation through production implementation and ongoing improvement.

The company develops intelligent automation solutions using AI and machine learning for large-scale document processing. It offers a highly autonomous engineering environment with a focus on continuous improvement and innovation.

US UK Unlimited PTO

  • Design and optimize training and post-training pipelines for large language models.
  • Improve model quality through supervised fine-tuning, reinforcement learning, and evaluation.
  • Build PyTorch-based training infrastructure and optimize distributed training across multi-GPU environments.

Lightning AI builds an end-to-end platform for developing, training, and deploying AI systems, founded in 2019. They are a global company with offices in New York, San Francisco, Seattle, and London, backed by major venture capital firms, and foster a builder culture that values urgency, ownership, and open communication.

Canada 6w PTO

  • Lead the evolution of personalized recommendation systems using machine learning.
  • Build, deploy, and optimize machine learning models in production.
  • Collaborate with cross-functional teams to deliver AI-driven experiences.

The company builds advanced machine learning solutions to improve user discovery and engagement. It operates as a fully remote global team with a focus on collaboration and innovation.

$80,000–$120,000/yr
Global Unlimited PTO

  • Design, build, and operate high-load distributed backend services powering the company's ML infrastructure.
  • Take end-to-end ownership of core ML services and data pipelines from design to deployment and continuous improvement.
  • Partner with ML and product teams to understand their needs and turn them into reliable, reusable platform capabilities.

Constructor is an AI-first e-commerce search and discovery platform that helps shoppers find products and enables brands to drive revenue. The company is fully remote, diverse, and values ownership and collaboration.

$216,700–$303,400/yr
US

  • Design, develop, and deploy ML models, including large language models, for various NLP tasks.
  • Collaborate with cross-functional teams to gather requirements, define architectures, and iterate on model development.
  • Stay up-to-date with latest research and contribute to best practices for responsible ML development.

Reddit is a community of communities built on shared interests, passion, and trust, hosting the most open and authentic conversations on the internet. With over 100,000 active communities and approximately 126 million daily active users, it is one of the internet's largest sources of information, fostering a culture of authenticity and community.

Global 4w PTO

  • Build and improve production recommendation and ranking systems for eCommerce.
  • Develop personalization models across the customer journey, from candidate generation to ranking.
  • Design and run experiments, A/B tests, and scalable ML services.

Sweed builds an all-in-one cannabis retail platform combining POS, eCommerce, Marketing, Analytics, and Inventory Management. With over 200 team members, we're a remote-first, product-driven startup focused on simplicity, efficiency, and innovation.

US

  • Lead the ML strategy and architecture for Shopping Ads delivery across targeting, retrieval, ranking, conversion, and value optimization.
  • Own end-to-end model development from opportunity sizing to monitoring and iteration.
  • Drive complex initiatives requiring coordinated changes across multiple teams.

Reddit is a community of communities built on shared interests, passion, and trust, hosting open and authentic conversations. With 100,000+ active communities and 130 million daily active users, it is one of the internet's largest information sources.

$140,000–$150,000/yr
United States Canada

  • Design and maintain ML model productionization infrastructure for high-visibility product features.
  • Collaborate with data science to streamline model training, validation, and deployment.
  • Implement robust monitoring and alerting for model performance, drift, and data quality.

The Athletic is a sports media company powered by one of the largest global newsrooms in sports, delivering in-depth coverage of professional and college teams across North America and Europe. With over 500 full-time staff, they foster a collaborative culture focused on high-quality journalism and data-driven innovation.

Global 5w PTO

  • Contribute to the entire development cycle of cutting-edge large deep learning models, from dataset preparation to deployment.
  • Collaborate with engineers and researchers to translate research into practical applications, wearing many hats.
  • Work at an exciting moment to build something from the ground up with a kind and collaborative team.

Reka builds useful multimodal AI to empower organizations, as a globally distributed foundation model startup headquartered in San Francisco Bay Area. The team, including contributors from Google DeepMind and FAIR, embraces a remote-first culture and collaborates globally.

$165,000–$330,000/yr
US Unlimited PTO

  • Partner directly with customers to architect, build, and deploy high-scale production AI applications on Baseten's platform.
  • Own the journey from initial exploration to production deployment, translating ambiguous goals into reliable services.
  • Work across product, software development, performance engineering, and customer-facing implementations.

Baseten powers mission-critical inference for dynamic AI companies like Cursor and Notion. They are rapidly growing, recently raised a $1.5B Series F, and foster a collaborative, forward-thinking culture.

$270,000–$290,000/yr
US Unlimited PTO

  • Design, develop, and deploy machine learning and statistical models for forecasting and risk assessment.
  • Use experimentation and statistical methods to test product changes and improve customer experience.
  • Analyze large datasets to identify signals, engineer features, and inform model design.

Pipe is a financial technology company that provides a platform for businesses to access capital. It is a fully remote company with a culture of authenticity, humility, and excellence.

$148,500–$174,750/yr
United States

  • Design, build, and operate large-scale ML solutions for ad delivery performance.
  • Collaborate with cross-functional teams to deploy models into production.
  • Mentor peers and contribute to ML best practices and AI-native development.

The company is a technology firm focused on AI-driven advertising solutions. It operates in a remote-first environment with a collaborative and inclusive culture.

Global 6w PTO 26w maternity 26w paternity

  • Manage the program portfolio covering inference, efficiency, serving, and endpoints to scale Cohere's infrastructure.
  • Lead cross-functional coordination with Modeling and customer-facing teams for end-to-end execution.
  • Identify pain points, establish processes, and improve engineering best practices.

Cohere is the leading security-first enterprise AI company building cutting-edge foundation AI models and end-to-end products. We are a global team of researchers, engineers, and designers passionate about our craft, with offices across North America and Europe.

US Unlimited PTO

  • Architect and build AI platforms, working across distributed teams to deliver GenAI products like Abacus Agents, MCP, and RAG systems.
  • Develop ML/AI pipelines for data preprocessing, model training, and evaluation, enabling rapid experimentation and self-healing systems.
  • Mentor software developers, participate in architectural discussions, and communicate technical direction to align teams and stakeholders.

Abacus Insights transforms healthcare data for health plans to improve outcomes, reduce waste, and enhance member and provider experiences. Backed by $100M from top investors, the company is a collaborative, curiosity-driven team of bold individuals embracing AI and automation to innovate in the healthcare industry.