Software Engineer - Machine Learning Infrastructure (Remote)

About the Role

We are seeking a Software Engineer - Machine Learning Infrastructure (Remote) to join our dynamic team at Stripe. In this role, you will have the opportunity to work on cutting-edge machine learning technologies and contribute to our mission of increasing the GDP of the internet. As a key member of the ML Infra team, you will help build the foundational infrastructure that powers machine learning across our organization.

What You'll Do

• Design and build scalable, reliable, and secure services for notebooks, ML model training, experimentation, serving, and LLM applications across multiple regions.
• Create services and libraries that enable ML engineers at Stripe to seamlessly transition from experimentation to production across Stripe’s systems.
• Work directly with product teams and ML engineers to improve their day-to-day productivity.
• Take ownership of and find solutions for technical and product challenges by working with a diverse set of systems, processes, and technologies.
• Collaborate with machine learning engineers, data scientists, and product engineering teams to enable a seamless end-to-end experience in building solutions across data, analytics, and AI/ML platforms.

Requirements

• 2+ years of professional software development experience with a solid background in service-oriented architecture and large-scale distributed systems.
• Experience working through the full life cycle of software development, from talking to users, to design and implementation, to testing and deployment, to operations.
• Experience working on production ML platforms, MLOps solutions, or building LLM applications.
• Experience running operations for high availability, low latency systems.
• A sense of pragmatism: knowing when to aim for the ideal solution and when to adjust course.

Nice to Have

• Experience building and shipping production AI agents.
• Familiarity with LLMs and LLM frameworks.
• Experience training and shipping machine learning models to production to solve critical business problems.

What We Offer

• Competitive salary and benefits package.
• Remote work flexibility.
• Opportunities for professional growth and development.
• A collaborative and inclusive work environment.
• Access to cutting-edge technologies and tools.