Machine Learning Engineer (LLM inference)

GMI Cloud; Mountain View (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

GMI Cloud, a fast-growing AI infrastructure company, is hiring a Machine Learning Engineer, LLM Optimization to build a world-leading inference optimization team. You will drive research, validation, and productionization of advanced optimization techniques to boost latency, throughput, and cost efficiency across the inference platform.

You will focus on B200-first optimization with H200 evolution, collaborating with platform and infra teams to turn new ideas into measurable customer-ready

Qualifications

• Experience with LLM inference systems and performance optimization.
• Familiar with inference metrics and tradeoffs (throughput, latency, memory).
• Hands-on with GPU-based inference and model serving architectures.

Responsibilities

• Drive frontier research and engineering in LLM inference optimization for high performance.
• Develop optimization strategies for large-scale LLM serving across runtimes and production platforms.
• Advance quantization, speculative decoding, KV cache, and memory optimization techniques.

Skills

LLM inference

Python

GPU inference

Performance profiling

System optimization

Benchmarks

observability

Cross-functional collaboration

Tools

SGLang

vLLM

TensorRT-LLM

Triton

Job description

GMI Cloud is a fast-growing AI infrastructure company backed by Headline VC and one of only six cloud providers worldwide to earn NVIDIA’s prestigious Reference Platform Cloud Partner designation . We operate 8 of our own GPU clusters across the U.S. and Asia, delivering a full spectrum of services from GPU compute service to AI model inference API solutions. As an NVIDIA Reference Platform Cloud Partner, our infrastructure meets the highest standards for performance, security, and scalability in AI deployments. We empower AI startups and enterprises to “build AI without limits,” providing everything they need to prototype, train, and deploy AI models quickly and reliably.

About this role

We are hiring a Machine Learning Engineer, LLM Optimization to build a world-leading inference optimization team and make GMI Cloud the industry benchmark for LLM serving performance.

This role is for engineers who want to work at the frontier of AI systems. You will drive the research, validation, and productionization of the most advanced inference optimization techniques, and turn them into real competitive advantage across GMI’s inference platform.

Our goal is to make GMI the company that leads the industry in how fast we discover, evaluate, combine, and operationalize the best optimization strategies for real customer workloads. That means not only adopting the latest advances, but also defining best practices, developing our own optimization methodologies, and building the internal framework that keeps GMI ahead of the curve.

You will focus on B200-first optimization, with support for H200 evolution, across core domains including quantization, speculative decoding, KV cache and memory management, prefill/decode disaggregation, and system-level inference optimization. You will work closely with platform and infrastructure teams to transform cutting-edge ideas into measurable gains in latency, throughput, cost efficiency, and production scalability.

Key Responsibilities

• Drive frontier research and engineering in LLM inference optimization, building GMI’s industry-leading capabilities in performance, efficiency, and scalability.
• Develop next-generation optimization strategies for large-scale LLM serving across model execution, runtime systems, and production inference platforms.
• Advance state-of-the-art techniques in quantization and precision optimization to improve throughput, latency, memory efficiency, and cost-performance across modern GPU systems.
• Push the frontier of speculative decoding and related acceleration methods, including both systems and model-level approaches for faster generation.
• Lead innovation in KV cache and memory optimization, improving long-context serving efficiency, memory utilization, and multi-tenant performance.
• Develop advanced architectures for prefill/decode disaggregation and other distributed inference optimization strategies for large-scale production environments.
• Drive system-level optimization across scheduling, batching, routing, gateway orchestration, adapter serving, and end-to-end inference efficiency.
• Build scalable optimization frameworks, performance methodologies, and engineering practices that allow GMI to stay ahead of the industry as models, hardware, and serving patterns evolve.
• Turn cutting-edge optimization ideas into production-ready capabilities that improve real-world customer workloads across latency, throughput, quality, and cost.
• Collaborate closely with platform, infrastructure, and product teams to make inference optimization a core technical advantage of GMI Cloud.

Required Skills

• Strong hands-on experience with LLM inference systems and performance optimization.
• Solid understanding of inference metrics and tradeoffs, including TTFT, ITL, throughput, goodput, tail latency, GPU utilization, memory efficiency, and quality/cost tradeoffs.
• Experience with one or more modern serving stacks such as SGLang, vLLM, TensorRT-LLM, Triton, or similar systems.
• Deep familiarity with GPU-based inference, model serving architecture, and production bottlenecks around compute, memory bandwidth, KV-cache behavior, and scheduling.
• Strong experimentation skills: able to design benchmarks, interpret results, debug regressions, and produce actionable conclusions rather than isolated microbenchmark wins.
• Comfortable working across research-style validation and production engineering, with a bias toward measurable impact in real customer scenarios.
• Strong coding and systems skills in Python, with practical experience in profiling, observability, and performance debugging.
• Clear communication skills and the ability to explain technical tradeoffs to both engineers and cross-functional stakeholders.

Preferred Qualifications

• 1+ years of hands-on experience in LLM inference optimization, ML systems optimization, or closely related areas.
• Experience working on optimization for large-scale model serving, such as latency reduction, throughput improvement, memory efficiency, or cost-performance tuning.
• Familiarity with one or more major areas of inference optimization, including quantization, speculative decoding, KV cache optimization, prefill/decode disaggregation, or system-level serving optimization.
• Experience with modern LLM serving stacks, GPU inference systems, or production ML infrastructure is a strong plus.