Staff ML Platform Engineer — AI Infra Architect

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

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.