Skip to content
AbridgeAbridgeSan Francisco, CA

Machine Learning Infrastructure Engineer- Model Inference

Builds and optimizes scalable ML inference infrastructure using Kubernetes and GPU resources to deploy production AI models with low latency. Collaborates with ML research and product teams on model serving, orchestration, and compute efficiency.

221k – 260k/yr
HybridML Engineering

About the role

What You'll Do

  • Design, deploy and maintain scalable Kubernetes clusters for AI model inference and training
  • Develop, optimize, and maintain ML model serving infrastructure, ensuring high-performance and low-latency.
  • Collaborate with ML and product teams to scale backend infrastructure for AI-driven products, focusing on model deployment, throughput optimization, and compute efficiency.
  • Optimize compute-heavy workflows and enhance GPU utilization for ML workloads.
  • Build a robust model API orchestration system
  • Collaborate with leadership to define and implement strategies for scaling infrastructure as the company grows, ensuring long-term efficiency and performance.

What You’ll Bring

  • Strong experience in building and deploying machine learning models in production environments.
  • Deep understanding of container orchestration and distributed systems architecture
  • Expertise in Kubernetes administration, including custom resource definitions, operators, and cluster management
  • Experience developing APIs and managing distributed systems for both batch and real-time workloads
  • Excellent communication skills, with the ability to interface between research and product engineering

Ideally, You Have

  • Expertise with model serving frameworks such as NVIDIA Triton Server, VLLM, TRT-LLM and so on.
  • Expertise with ML toolchains such as PyTorch, Tensorflow or distributed training and inference libraries.
  • Familiarity with GPU cluster management and CUDA optimization
  • Knowledge of infrastructure as code (Terraform, Ansible) and GitOps practices
  • Experience with container registries, image optimization, and multi-stage builds for ML workloads
  • Experience orchestrating across ASR models or LLM models for building various GenAI applications

Skills

KubernetesPyTorchTensorFlowNvidia Triton ServervLLMTrt-LlmTerraformAnsibleCUDAGPUDistributed SystemsAPIs

Similar roles

ML Engineering jobs
Perplexity

AI Engineer, Applied ML

PerplexitySan Francisco, CA +2

Designs, builds, and iterates on AI/ML models for personalization, query understanding, and content discovery. Requires 5+ years in ML, deep learning expertise (PyTorch/TensorFlow/JAX), Python, and full ML lifecycle ownership.

220k – 405k/yr
On-site5+ YOEML Engineering
Perplexity

AI Inference Engineer

PerplexitySan Francisco, CA +2

Develops APIs and optimizes large-scale ML model inference using Python, Rust, C++, PyTorch, and CUDA. Benchmarks performance, improves reliability, and implements LLM optimizations on GPU architectures.

220k – 405k/yr
On-siteML Engineering
Scale AI

ML Systems Engineer, Robotics

Scale AISan Francisco, CA

Designs and builds scalable platforms for serving foundation models in robotics and autonomous vehicles. Requires 4+ years in ML infrastructure, backend systems, GPU optimization, and cloud technologies.

218k – 273k/yr
On-site4+ YOEML Engineering
Scale AI

ML Research Engineer, ML Systems

Scale AISan Francisco, CA +2

Builds and optimizes distributed frameworks for LLM training and inference on Scale's RLXF platform. Collaborates with ML teams to accelerate research, requiring expertise in PyTorch, CUDA, transformers, and large-scale distributed systems.

218k – 273k/yr
On-siteML Engineering
Scale AI

Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI

Scale AISan Francisco, CA +2

Develops synthetic data pipelines, production trace agents, and automated agent-building frameworks for enterprise GenAI. Requires 3+ years LLM production experience, top conference publications, and advanced CS degree.

218k – 273k/yr
Hybrid3+ YOEML Engineering