7mo ago

Member of Technical Staff, Inference & Serving

Bay Areafull-timelead

OtherMember Of Technical Staff

0 views0 saves0 applied

Apply Now

Quick Summary

Overview

The Role We're looking for engineers and scientists to design, optimize, and scale the systems that power our diffusion LLMs in production. Your work will make inference faster, more cost-effective,

Technical Tools

OtherMember Of Technical Staff

The Role

We're looking for engineers and scientists to design, optimize, and scale the systems that power our diffusion LLMs in production. Your work will make inference faster, more cost-effective, and more reliable.

Key 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 (new architectures, quantization techniques, batching strategies) into production-ready serving improvements.

Qualifications

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

Preferred Skills

Experience building and maintaining large-scale language models with tens of billions of parameters or more.
Experience with distributed systems and cloud computing platforms (AWS/GCP/Azure).
Experience with ML workflow orchestration tools (Kubeflow, Airflow).
Experience with model optimization techniques (quantization, distillation, speculative decoding, continuous batching).
Knowledge of ML-specific infrastructure challenges (checkpointing, resource scheduling, etc.).

What We Offer

~1 min read

The annual base salary range for this role is $200,000 – $350,000 USD. Final compensation is determined based on experience, skills, and qualifications. Equity and benefits are included in the total package.