Senior Software Engineer, ML/AI Platform

United States Digital Space LLC; United States; On-site; USD 180,000 - 250,000; Full time

Benefits offered by this job

Health benefits

Equity

Wellness programs

Job summary

United States Digital Space LLC is seeking a Senior Software Engineer to join the Machine Learning Platform team. You will build and operate end-to-end ML workflows—from data access and training to deployment and serving—ensuring scalable, reliable AI capabilities for the business.

You will collaborate with ML, Data Science, Product Engineering, and Infrastructure teams, delivering self-service capabilities and robust observability.

Qualifications

• 5+ years building and operating production software or distributed systems, with ML platform or ML infrastructure experience.
• Strong software engineering skills in Python or Java and building production services.
• Experience across ML lifecycle areas: training infrastructure, deployment, monitoring, or inference.

Responsibilities

• Build and operate production-grade services and workflows for training, evaluating, deploying, and serving ML models.
• Improve ML data access, training infrastructure, model tooling, and inference pipelines.
• Develop scalable APIs and self-service tools for ML engineers and data scientists.
• Improve reliability, observability, performance, and cost efficiency of ML platform components.
• Own projects from design through production, with cross-functional collaboration.
• Contribute to designs and break down platform initiatives into milestones.
• Partner with ML, Data Science, Product Engineering, and Infrastructure teams to deliver AI initiatives.
• Raise engineering quality through reviews, documentation, and best practices.

Skills

Python

Java

ML platform

Kubernetes

AWS

Distributed systems

Tools

Terraform

Datadog

Metaflow

MLflow

Argo

PyTorch

TensorFlow

Hugging Face

Spark

Kafka

Ray

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 looking for a self-motivated, highly driven Senior Software Engineer to join the company’s Machine Learning Platform team. As a hands-on individual contributor, you’ll build and operate the platform capabilities that enable ML engineers and data scientists to develop, train, evaluate, deploy, and serve models reliably at scale.

Your work will span the ML lifecycle—including data and feature access, training infrastructure, model lifecycle management, batch and real-time inference, and platform observability. You’ll own well-scoped projects from design and implementation through production operations, partnering with Staff engineers and platform users to translate broader architectural direction into reliable, self-service capabilities.

This is an opportunity to join a growing ML Platform team and directly improve how quickly and safely the company ships AI- and ML-powered products.

What You’ll Accomplish

• Build and operate production-grade services and workflows for training, evaluating, deploying, and serving ML models.
• Implement improvements across ML data and feature access, training infrastructure, model lifecycle tooling, and batch and online inference.
• Develop scalable APIs, abstractions, and self-service tools that make ML engineers and data scientists more productive.
• Improve the reliability, observability, performance, and cost efficiency of ML platform components.
• Own projects through implementation, testing, rollout, monitoring, and ongoing production support.
• Contribute to technical designs and help break larger platform initiatives into deliverable milestones.
• Partner with ML, Data Science, Product Engineering, and Infrastructure teams to deliver AI and ML initiatives.
• Raise engineering quality through code reviews, documentation, operational best practices, and mentorship.

Your Expertise

• 5+ years building and operating production software or distributed systems, with meaningful experience in ML platform, MLOps, or ML infrastructure.
• Strong software engineering skills in Python, Java, or a comparable language, including experience building production services or developer-facing platforms.
• Experience in one or more areas of the ML lifecycle: training infrastructure, orchestration, feature platforms, model deployment, model lifecycle management, or inference.
• Experience with distributed data or compute technologies such as Spark, Ray, Kafka, or similar systems. Expertise in every listed technology is not required.
• Understanding of the design and operational tradeoffs between batch, streaming, online, and offline ML workloads.
• Experience running reliable workloads in a cloud environment using technologies such as Kubernetes, AWS, and infrastructure-as-code.
• Ability to diagnose performance, scalability, and reliability problems across application, data, and infrastructure layers.
• A track record of independently delivering complex projects while collaborating effectively with technical and business partners.

What We Use

• AWS, EKS, Kubernetes, Terraform, Helm, Istio, and Datadog
• Metaflow, MLflow, Argo, PyTorch, TensorFlow, and Hugging Face
• Python and Java services
• DynamoDB, Postgres, Redis, Kinesis, and other production data systems
• Spark, Ray, Kafka, and distributed batch and streaming systems

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 $180,000 - $250,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 AWAKE, the company’s collective of employee resource groups.

If you do not meet all the requirements listed here, we still encourage you to apply! No job description is perfect, and we may also have another opportunity that closely matches your skills and experience.

At the company, we know that our Company's strength lies in the diversity of our employees. the company is an Equal Opportunity Employer and we welcome applicants from all backgrounds. Our policy is to provide equal employment opportunities for all employees, applicants and covered individuals regardless of protected characteristics. We prioritize and maintain a fair, inclusive and equitable workplace free from discrimination, harassment, and retaliation. the company is also committed to providing reasonable accommodations for candidates with disabilities. If you need any assistance or reasonable accommodations, please let your recruiter know.