Staff ML Systems Engineer - Diffusion LLM Serving

Inception; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Inception is seeking engineers and scientists to design, optimize, and scale the diffusion LLM serving systems powering production inference. Your work will help make inference faster, more cost-efficient, and more reliable.

You will extend orchestration frameworks (Kubernetes, Ray, SLURM) for distributed inference, evaluation, and large-batch serving, and implement load balancing, autoscaling, and traffic routing for model endpoints.

Qualifications

• BS/MS/PhD in CS/Engineering or related field.
• Knowledge of ML serving frameworks (SGLang, vLLM, Triton Inference Server, TensorRT-LLM).
• Understanding of ML frameworks (PyTorch, TensorFlow) from a systems perspective.
• Familiarity with HPC and GPU programming (CUDA).
• Experience with containers (Docker), orchestration (Kubernetes), and CI/CD pipelines.
• Background in performance optimization and profiling of ML systems.

Responsibilities

• Build and optimize high-performance model serving systems for low-latency inference of diffusion LLMs.
• Extend orchestration frameworks (Kubernetes, Ray, SLURM) for distributed inference, evaluation, and large-batch serving.
• Implement and manage load balancing, autoscaling, and traffic routing for model endpoints.
• Build systems for model versioning, canary deployments, and zero-downtime rollouts.
• Develop monitoring, alerting, and observability tooling to ensure SLA compliance and rapid incident response.
• Collaborate with ML researchers to translate model advances into production-ready serving improvements.

Skills

Distributed inference

Performance optimization

Profiling ML systems

Collaboration with ML researchers

Education

BS/MS/PhD in CS/Engineering

Tools

Docker

Kubernetes

CI/CD

CUDA

PyTorch

TensorFlow

Kubeflow

Airflow

SGLang

vLLM

Triton Inference Server

TensorRT-LLM

Job description

Inception is seeking engineers and scientists to design, optimize, and scale the diffusion LLM serving systems powering production inference. Your work will help make inference faster, more cost-efficient, and more reliable.

You will extend orchestration frameworks (Kubernetes, Ray, SLURM) for distributed inference, evaluation, and large-batch serving, and implement load balancing, autoscaling, and traffic routing for model endpoints.