Staff Machine Learning Platform Engineer

United States Digital Space LLC; United States; Hybrid; USD 156,000 - 215,000; Full time

Job summary

United States Digital Space LLC is seeking a Staff Machine Learning Platform Engineer to design, operate, and scale a production ML platform supporting thousands of local businesses.

You will bridge data science and production engineering, building scalable workloads and governance while collaborating with multiple ML teams across a shared platform environment. The role emphasizes MLOps, cloud infrastructure, and process improvements.

Qualifications

• 8+ years of experience building production ML or data platforms
• A degree (preferably graduate level) in Computer Science, Engineering, Statistics, or a related technical field
• Strong hands‑on expertise with Databricks, Spark, Delta Lake, and MLflow.
• Proficiency in Python, SQL, and distributed systems concepts
• Experience with cloud platforms and infrastructure-as-code
• Solid understanding of MLOps best practices: CI/CD, monitoring, reproducibility, and security
• Experience supporting multiple ML teams in a shared platform environment
• Active owner of orphaned problems and willingness to assimilate missing knowledge to get the job done

Responsibilities

• Design and operate ML infrastructure, including workspaces, clusters, jobs, and workflows
• Productionize ML workloads using Spark, Delta Lake, MLflow, and Databricks Workflows
• Teach data scientists how to utilize our ML platform to advance development from notebook to production for our most critical models
• Implement Unity Catalog for data governance, lineage, access control, and secure multi-tenant usage
• Build CI/CD pipelines for ML using Terraform and Git-based workflows (e.g., GitHub Actions)
• Optimize performance, reliability, and cost across training and inference workloads
• Configure Identity and Access Management (IAM) and Role Based Authentication Controls (RBAC) for sensitive data sets
• Establish observability for data quality, model performance, and platform health
• Build and maintain ML Platform technical documentation

Skills

Databricks

Spark

Delta Lake

MLflow

Python

SQL

Cloud platforms

IaC

MLOps

Multi-team support

Education

Bachelor's degree or higher in CS/Engineering/Statistics or related field

Tools

Terraform

GitHub Actions

Kubernetes

Snowflake

Job description

About the company

the company is a technology wholesale platform built on the belief that the future is local. Independent retailers around the globe collectively represent a multi-hundred-billion-dollar wholesale market that has historically been fragmented and offline. At the company, we're using the power of tech, data, and machine learning to connect this thriving community of entrepreneurs across the globe. Picture your favorite boutique in town — we help them discover the best products from around the world to sell in their stores. With the right tools and insights, we believe that we can level the playing field so businesses can grow and local communities can thrive.

We’re looking for smart, resourceful and passionate people to join us as we power the shop local movement. If you believe in community, come join ours.

Note:

This role can also be performed remotely for candidates located in Ontario, Canada.

About this role

As a Staff Machine Learning Platform Engineer, you will help design, improve, and operate a scalable ML platform to accelerate model training, deployment, and governance. You are the technical bridge between data science and production engineering. You’ll be joining a small but deeply critical team that scales the company’s ability to support tens of thousands of local businesses in a constantly narrowing retail landscape.

What You Will Do

• Design and operate ML infrastructure, including workspaces, clusters, jobs, and workflows
• Productionize ML workloads using Spark, Delta Lake, MLflow, and Databricks Workflows
• Teach data scientists how to utilize our ML platform to advance development from notebook to production for our most critical models
• Implement Unity Catalog for data governance, lineage, access control, and secure multi-tenant usage
• Build CI/CD pipelines for ML using Terraform and Git-based workflows (e.g., GitHub Actions)
• Optimize performance, reliability, and cost across training and inference workloads
• Configure Identity and Access Management (IAM) and Role Based Authentication Controls (RBAC) for sensitive data sets
• Establish observability for data quality, model performance, and platform health
• Build and maintain ML Platform technical documentation

What it takes

• 8+ years of experience building production ML or data platforms
• A degree (preferably graduate level) in Computer Science, Engineering, Statistics, or a related technical field
• Strong hands‑on expertise with Databricks, Spark, Delta Lake, and MLflow.
• Proficiency in Python, SQL, and distributed systems concepts
• Experience with cloud platforms and infrastructure-as-code
• Solid understanding of MLOps best practices: CI/CD, monitoring, reproducibility, and security
• Experience supporting multiple ML teams in a shared platform environment
• Are an active owner of orphaned problems and are willing to assimilate whatever knowledge you’re missing to get the job done

Tech Stack

the company uses a modern cloud based tech stack. For this role, you’ll want to be proficient with the following:

CategoryTechnologiesLanguages

Python, SQL, Kotlin

ML Frameworks

PyTorch, MLFlow

Big Data & Processing

Spark, Kafka, Databricks, Snowflake, Fivetran, Iceberg, Unity Catalog, Datadog, Airflow, Cockroach DB, MySQL

Cloud & Infrastructure

AWS, S3, SageMaker, Kubernetes, Docker, GitHub Actions, Terraform

Generative AI

Claude Sonnet 4.5, ChatGPT 5.2

Salary Range

Canada: the pay range for this role is $216,000 to $297,000 per year.

This role will also be eligible for equity and benefits. Actual base pay will be determined based on permissible factors such as transferable skills, work experience, market demands, and primary work location. The base pay range provided is subject to change and may be modified in the future.

the company uses Artificial Intelligence (AI) to screen and select applicants for this position.

This job posting is for an existing vacancy.

Hybrid the company employees currently go into the office 3 days per week on Tuesdays, Thursdays, and a third flex day of their choosing (Monday, Wednesday, or Friday). Additionally, hybrid in-office roles will have the flexibility to work remotely up to 4 weeks per year. Specific Workplace and Information Technology positions may require onsite attendance 5 days per week as will be indicated in the job posting.

Why you’ll love working at the company

• Move fast: You'll own meaningful problems that serve customers around the globe with the agency to move fast and see your results clearly.
• Equipped to scale: We invest in what matters, including the latest enterprise AI tools, to help you work smarter and get more out of every day.
• Best in class: Our team is full of sharp, kind, and generous colleagues who care about their craft and about helping you grow in yours.
• Real rewards. Competitive pay, equity, and comprehensive benefits designed to support your life inside and outside of work.
• Belonging: We're intentional about building an environment where every the company employee has equal access to opportunities, growth, and success.

the company was founded in 2017 by a team of early product and engineering leads from Square. We’re backed by some of the top investors in retail and tech including: Y Combinator, Lightspeed Venture Partners, Forerunner Ventures, Khosla Ventures, Sequoia Capital, Founders Fund, and DST Global. We have headquarters in San Francisco and Kitchener-Waterloo, and a global employee presence across offices in Toronto, London, and New York. To learn more about the company and our customers, you can read more on our blog.

the company provides equal employment opportunities (EEO) to all employees and applicants for employment without regard to race, color, religion, sex, national origin, age, disability, genetics, sexual orientation, gender identity or gender expression.

the company is committed to providing access, equal opportunity and reasonable accommodation for individuals with disabilities in employment, its services, programs, and activities. Accommodations are available throughout the recruitment process and applicants with a disability may request to be accommodated throughout the recruitment process. We will work with all applicants to accommodate their individual accessibility needs. To request reasonable accommodation, please fill out our Accommodation Request Form (https://bit.ly/the company-form)

Privacy

For information about the type of personal data the company collects from applicants, as wel