Staff Software Engineer, ML/AI Platform

United States Digital Space LLC; United States; On-site; USD 200,000 - 300,000; Full time

Benefits offered by this job

Equity

Health & wellness benefits

Job summary

United States Digital Space LLC is seeking a Staff Software Engineer to join the Machine Learning Platform team. You will architect and build foundational AI infrastructure enabling training, deployment, and serving of models at scale.

You’ll lead technically, influence architectural directions, and mentor ML engineers. This role emphasizes building robust, low-latency systems and bridging platform capabilities with product needs across the organization.

Qualifications

• 7+ years focused specifically on ML Platform/MLOps.
• Proven track record of owning and building core components of ML platforms using tools like Spark, Ray, MLFlow, Kubeflow, or Metaflow.
• Strong expertise in Python for batch processing and online service frameworks.
• Experience designing and operating online and offline inference systems; understands their tradeoffs.

Responsibilities

• Setting Technical Direction - Architect ML platform strategy spanning data pipelines, training infrastructure, and serving layers using cutting-edge tooling like Ray, MLFlow, Metaflow, Argo, and Spark.
• Uplevel and Innovate Core AI & ML Stack - Build and operate production-grade, low-latency ML serving layers with robust model lifecycle systems including champion/challenger testing, automated rollouts, versioning, and rollback capabilities.
• Uplevel and Innovate Core AI & ML Stack - Define and drive the company’s agentic stack.
• Technical Leadership - Provide ML infrastructure perspective in high-level discussions about the company’s AI strategy spanning multiple quarters and teams.
• Technical Mentorship - Mentor platform and ML engineers, actively championing team members.
• Being the “Glue” - Build universal interfaces, architectures, and patterns—like data access layers and prediction serving APIs that bridge platform capabilities with product needs to streamline high-priority ML work across the organization

Skills

ML Platform

MLOps

Python

Technical Leadership

Mentorship

Tools

Spark

Ray

MLFlow

Kubeflow

Metaflow

Pytorch

TensorFlow

Pandas

Kubernetes

AWS

Job description

the company® is the AI marketing platform for 1:1 personalization redefining the way brands and people connect. We’re the only marketing platform that combines powerful technology with human expertise to build authentic customer relationships. By unifying SMS, RCS, email, and push notifications, our AI-powered personalization engine delivers bespoke experiences that drive performance, revenue, and loyalty through real-time behavioral insights.

Recognized as the#1provider in SMS Marketing by G2, the company partners with more than 8,000 customers across 70+ industries. Leading global brands like Crate and Barrel, Urban Outfitters, and Carter’s work with us to enable billions of interactions that power tens of billions in revenue for our customers.

With a distributed global workforce and employee hubs in New York City, San Francisco, London, and Sydney, the company’s team has been consistently recognized for its performance and culture. We’re proud to be included inDeloitte’s Fast 500(four years running!),LinkedIn’s Top Startups,Forbes’ Cloud 100 (five years running!),Inc.’s Best Workplaces, and theHuman Rights Campaign Foundation's Corporate Equality Index!

About the Role

We’re seeking an accomplished Staff Software Engineer to join the company’s Machine Learning Platform team as a high-impact individual contributor focused on building the AI and ML infrastructure that powers our AI product suite. You’ll architect and build the foundational platform components that enable AI / ML engineers and data scientists to train, deploy, and serve models and agentic infrastructure with velocity, performance, and reliability at scale.

As a Staff-level IC, you’ll operate as a technical force multiplier, setting the technical direction for AI and ML infrastructure across the company’s AI organization. You’ll lead through influence and technical excellence, advocating for long-term architectural progress while balancing immediate platform needs. Your work will span strategic initiatives measured in quarters and years, focusing on high-leverage decisions that enable entire teams to ship AI and ML capabilities faster and more reliably.

Strategic Needthe company is revolutionizing the digital shopping experience across every channel through our AI product suite for half a billion subscribers. We’re looking for a high impact individual to take our platform from v1 to vNext and beyond — supporting the full spectrum of AI and ML workloads at massive scale. We support traditional models and deep learning today, and we are growing into reinforcement learning and agentic infrastructure quickly. This is a ground-floor opportunity to drive and influence the architectural roadmap for the company’s entire AI and ML ecosystem toward self-service workflows, real-time inference at scale, agentic capabilities, and robust model lifecycle management.

What You’ll Accomplish

• Setting Technical Direction - Architect ML platform strategy spanning data pipelines, training infrastructure, and serving layers using cutting-edge tooling like Ray, MLFlow, Metaflow, Argo, and Spark

• Uplevel and Innovate Core AI & ML Stack - Build and operate production-grade, low-latency ML serving layers with robust model lifecycle systems including champion/challenger testing, automated rollouts, versioning, and rollback capabilities

• Uplevel and Innovate Core AI & ML Stack - Define and drive the company’s agentic stack

• Technical Leadership - Provide ML infrastructure perspective in high-level discussions about the company’s AI strategy spanning multiple quarters and teams

• Technical Mentorship - Mentor platform and ML engineers, actively championing team members

• Being the “Glue” - Build universal interfaces, architectures, and patterns—like data access layers and prediction serving APIs that bridge platform capabilities with product needs to streamline high-priority ML work across the organization

Your Expertise

• You have the experience to know what works, what doesn’t, and why in AI and ML systems

• 7+ years focused specifically on ML Platform/MLOps, with deep understanding of gold-standard practices and best-in-class tooling

• Proven track record of owning and building core components of ML platforms using tools like Spark, Ray, MLFlow, Kubeflow, or Metaflow

• You’ve built and operated a high-throughput agentic stack (MCP / data infrastructure, context store, orchestration, and prompt layer)

• Strong expertise in Python for both batch processing and online service frameworks

• Experience designing and operating online and offline inference systems, understanding the critical differences and tradeoffs between them

Sample Projects

• Design and implement inference pipelines with champion/challenger shadow testing and automated model promotion

• Lead and scale the company’s agentic stack from the ground up

• Scale real-time feature streaming to handle low-latency, high-volume reinforcement learning workloads

• Build a universal data access layer and prediction serving interface that powers ML capabilities across the company’s product suite

What We Use

• Our backend is Java / Kotlin / Spring Boot microservices, built with Gradle, coupled with things like DynamoDB, Aurora, AirFlow, Postgres, and Redis, hosted via AWS

• Our infrastructure runs primarily in Kubernetes hosted in AWS’s EKS, using tooling like Istio, Datadog, Terraform, CloudFlare, and Helm

• Our frontend is built with React and TypeScript, and uses best practices like GraphQL, Storybook, Radix UI, Vite, esbuild, and Playwright

• Our automation is driven by custom and open source machine learning models, industry-leading LLMs, lots of data and tech like Python, Metaflow, HuggingFace, PyTorch, TensorFlow, and Pandas

You’ll get competitiveperks and benefits, from health & wellness to equity, to help you bring your best self to work.

For US based applicants:

• The US base salary range for this full-time position is $200,000 - $300,000 annually+ equity + benefits

• Our salary ranges are determined by role, level and location

#LI-EZ1

By applying for this position, your data will be processed as per the company's Privacy Policy.the company Company ValuesDefault to Action - Move swiftly and with purposeBe One Unstoppable Team - Rally as each other’s championsChampion the Customer - Our success is defined by our customers' successAct Like an Owner - Take responsibility for the company’s success

Learn more about