ML Platform Engineer: Scale AI Infra, Deploy & Optimize

United States Digital Space LLC; United States; Remote; USD 120,000 - 180,000; Full time

Job summary

United States Digital Space LLC is seeking an ML Platform Engineer to design and operate the infrastructure powering our AI capabilities. You will build platforms for model training, evaluation, deployment, inference, observability, and continuous improvement.

You will collaborate with AI researchers and product engineers to turn evolving model requirements into reliable, scalable production systems that deliver low latency, high throughput, and cost efficiency at scale.

Qualifications

• Strong software engineering fundamentals and production systems experience.
• Experience building ML infrastructure, platforms, or production machine learning systems.
• Experience with model deployment, inference, evaluation, or data pipelines.
• Strong understanding of distributed systems and system reliability.
• Ability to write clean, maintainable, production-quality code.
• Comfortable working in ambiguous, fast-moving environments.
• Bias toward ownership, experimentation, and continuous improvement.

Responsibilities

• Build and operate the ML infrastructure and platforms powering AI products.
• Design systems for model training, evaluation, deployment, inference, and experimentation.
• Build and optimise model serving and inference infrastructure for high-throughput and low-latency workloads.
• Improve reliability, scalability, latency, and cost efficiency of AI systems.
• Develop reliable pipelines for data preparation, training, evaluation, model release, and continuous improvement.
• Work closely with AI engineers, researchers, and product teams to turn evolving model requirements into production-ready infrastructure.

Skills

Software engineering fundamentals

ML infrastructure experience

Model deployment

Distributed systems

Tools

Python

PyTorch

JAX

LLM serving

Cloud infra

Distributed systems

ML pipelines

GPU tooling

Vector databases

Job description

United States Digital Space LLC is seeking an ML Platform Engineer to design and operate the infrastructure powering our AI capabilities. You will build platforms for model training, evaluation, deployment, inference, observability, and continuous improvement.

You will collaborate with AI researchers and product engineers to turn evolving model requirements into reliable, scalable production systems that deliver low latency, high throughput, and cost efficiency at scale.