Staff Machine Learning Engineer

Toast creates technology to help restaurants and local businesses succeed in a digital world, helping business owners operate, increase sales, engage customers, and keep employees happy.

The Machine Learning Platform team builds and operates the core infrastructure that powers ML across Toast — the feature store, model hosting and serving, the experimentation platform, training pipelines, and the tooling ML engineers and data scientists rely on every day. Our work directly enables the models that drive personalization, forecasting, fraud detection, search, and the growing set of AI-powered experiences shipping to restaurants.

Toast is seeking a Staff Software Engineer to act as a technical leader on the ML Platform team, shaping the systems that will carry Toast's ML capabilities into the next decade. The role involves driving architectural direction across the platform, delivering foundational infrastructure that other teams build on, and elevating fellow engineers. The ideal candidate is a domain expert who partners with ML engineers, data scientists, product, and infrastructure leadership on high-leverage opportunities.

This position suits an engineer comfortable writing production code, leading technical design for distributed systems, and influencing organizational decisions about how Toast builds and deploys ML.

A day in the life (Responsibilities)

Own technical direction of the ML Platform — feature store, model hosting and serving, experimentation, training infrastructure — driving architectural decisions around scalability, reliability, latency, and cost

Lead design and delivery of large-scope platform initiatives from conception through production, coordinating across ML, data, and infrastructure teams

Identify and resolve systemic technical challenges: online/offline feature parity, model deployment friction, experimentation velocity, GPU utilization, cross-team dependencies

Set and maintain a high engineering quality bar through hands-on code contributions, design reviews, and mentorship of platform and ML-adjacent engineers

Partner with ML engineering, data science, product, and platform leadership to translate ML strategy into technical roadmaps

Define the paved paths ML teams use to ship models safely — from feature registration through canary rollout, monitoring, and rollback

Leverage AI-augmented development tools to increase development velocity and code quality

What you'll need to thrive (Requirements):

8+ years delivering complex backend or infrastructure systems at scale

Direct experience building or operating core ML infrastructure — feature stores, model serving, experimentation platforms, training orchestration, or equivalent

Mastery of a modern backend language such as Python, Java, Kotlin, Go, or Scala

Deep proficiency with distributed systems concepts: consistency, latency, throughput, fault tolerance, and observability

Strong understanding of data modeling, query languages, and the online/offline data patterns that underpin ML systems

Demonstrated technical leadership, with ability to drive cross-team alignment and influence engineering, product, and business stakeholders

Bachelor's degree in Computer Science or a related field, or equivalent practical experience

Nice to Haves:

Hands-on experience with open-source or commercial ML platform components (e.g. Tecton, MLflow, SageMaker, Databricks)

Experience building or operating experimentation / A-B testing platforms at scale

Familiarity with real-time streaming systems (Kafka, Flink, Spark Streaming) and their use in feature computation

Experience serving LLMs or large deep-learning models in production, including GPU capacity planning and inference optimization

Comfort with Kubernetes and modern cloud-native infrastructure

Prior work supporting internal-developer-facing platforms with a product mindset

AI at Toast

At Toast, one of our company values is that we're hungry to build and learn. We believe learning new AI tools empowers us to build for our customers faster, more independently, and with higher quality. We provide these tools across all disciplines, from Engineering and Product to Sales and Support, and are inspired by how our Toasters are already driving real value with them. The people who thrive here are those who embrace changes that let us build more for our customers; it’s a core part of our culture.

Our Total Rewards Philosophy

We strive to provide competitive compensation and benefits programs that help to attract, retain, and motivate the best and brightest people in our industry. Our total rewards package goes beyond great earnings potential and provides the means to a healthy lifestyle with the flexibility to meet Toasters’ changing needs. Learn more about our benefits at https://careers.toasttab.com/toast-benefits .

#LI-REMOTE

The base salary range for this role is listed below. The starting salary will be determined based on skills, experience, and geographic location. In addi…