Senior ML Platform Engineer — AI Infra

United States Digital Space LLC; Bellevue (CA); On-site; USD 209,000 - 245,000; Full time

Benefits offered by this job

Health insurance

Equity ownership

401k matching

Catered meals

Mental health benefits

Job summary

United States Digital Space LLC is seeking a Senior Software Engineer for the AI Infrastructure team in Menlo Park, CA and Bellevue, WA. You will own architecture, build scalable ML platform components, and partner with ML practitioners to accelerate model development and deployment.

You will mentor engineers, shape technical strategy, and ensure reliable, observable systems powering our AI products in production.

Qualifications

• Bachelor's degree in Computer Science, Software Engineering, or a related technical field.
• Proven ability to own and deliver complex platform systems end-to-end, from architecture to production.
• Deep expertise in model serving, distributed systems, and production ML workflows at scale.

Responsibilities

• Lead the architecture and end-to-end delivery of scalable systems for deploying, monitoring, and managing ML models in production
• Own the technical direction for key platform areas — including model serving, the feature store, and ML observability infrastructure
• Drive cross-functional partnerships with ML practitioners, data engineers, and applied AI teams to streamline workflows
• Evolve and scale our feature store to support efficient, low-latency feature retrieval across real-time and batch use cases
• Define and implement robust observability standards for model performance, data pipelines, and feature freshness across the ML platform
• Manage and optimize cloud compute resources (CPU/GPU) on AWS to support cost-effective, high-throughput training and inference at scale
• Contribute to technical strategy and roadmap discussions, and help mentor engineers on the team through design reviews and hands-on guidance

Skills

ML infrastructure

Distributed systems

Python

C++

Model serving

TensorFlow

PyTorch

Kubeflow

SageMaker

Vector search

Education

Bachelor's degree in CS/CE/related

Advanced degree a plus

Tools

TensorFlow Serving

Ray

Kubeflow

SageMaker

Elasticsearch

Qdrant

ChromaDB

Job description

United States Digital Space LLC is seeking a Senior Software Engineer for the AI Infrastructure team in Menlo Park, CA and Bellevue, WA. You will own architecture, build scalable ML platform components, and partner with ML practitioners to accelerate model development and deployment.

You will mentor engineers, shape technical strategy, and ensure reliable, observable systems powering our AI products in production.