H

Hippocratic AI

Licenses safety-focused healthcare LLMs to institutions

LLM Inference Systems Engineer

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeMenlo Park, CA, USANo salary listed

About the Job

Requirements

  • Production experience building or operating distributed systems, high-performance computing systems, or large-scale machine learning inference platforms.
  • Strong understanding of large language model inference, including tensor, pipeline, and data parallelism, continuous batching, key-value cache management, and prefill/decode behavior.
  • Hands-on experience with GPU communication and networking technologies such as NCCL, NVLink/NVSwitch, InfiniBand, RoCE, RDMA, UCX, or equivalent systems.
  • Strong Python skills and working proficiency in C++ or another systems language.
  • Ability to profile and debug performance across application, runtime, kernel, network, and infrastructure layers.
  • Experience deploying production workloads on Kubernetes and operating them with clear reliability and observability standards.
  • Master's or PhD required.

Responsibilities

  • Design, implement, and operate disaggregated large language model serving architectures, including independent prefill and decode pools, key-value cache transfer, routing, batching, and failure recovery.
  • Optimize multi-node inference across modern GPU systems using NVLink, NVSwitch, NVLS, InfiniBand, RoCEv2, GPUDirect RDMA, NCCL, UCX, and related communication paths.
  • Build and tune serving systems using SGLang, vLLM, NVIDIA Dynamo, TensorRT-LLM, or comparable frameworks.
  • Profile end-to-end performance across compute, memory, network, and storage; improve time to first token, inter-token latency, throughput, tail latency, and cost per token.
  • Develop repeatable benchmark and capacity-planning workflows across model architectures, GPU types, parallelism strategies, and concurrency levels.
  • Productionize serving on Kubernetes with health checks, autoscaling, safe rollouts, metrics, tracing, and actionable diagnostics.
  • Diagnose complex distributed failures such as collective timeouts, topology mismatches, packet loss, congestion, key-value transfer stalls, GPU out-of-memory errors, and uneven load.
  • Partner with model researchers and platform engineers to launch new models, serving features, and hardware generations safely.

Desired Qualifications

  • Direct experience with disaggregated serving, key-value cache transfer, NVIDIA Dynamo/NIXL, SGLang, vLLM, or TensorRT-LLM.
  • Experience with NVLS, SHARP, GPUDirect RDMA, UCX, RDMA congestion control, or GPU-cluster topology optimization.
  • CUDA, Triton, custom kernel, or low-level GPU performance experience.
  • Experience with speculative decoding, multi-LoRA serving, quantization, or cache-aware routing.
  • Contributions to open-source inference, networking, or distributed-systems projects.
  • Experience benchmarking new GPU platforms and turning results into production architecture decisions.

About the Company

Hippocratic AI

Licenses safety-focused healthcare LLMs to institutions

Hippocratic AI builds specialized language models for healthcare that hospitals and clinics license to support non-clinical tasks and improve patient care and operations. The models are trained on evidence-based medical content and refined with a Reinforcement Learning with Human Feedback process from healthcare professionals to ensure safety and reliability. It differentiates itself by prioritizing safety, equity, and bedside-appropriate interactions, with validation on healthcare exams to ensure real-world applicability. Its goal is to improve healthcare access, equity, and outcomes by providing safe, trustworthy AI tools for clinicians and patients.

Company size
201-500
Company stage
Series C
Total funding
$387M
Headquarters
Palo Alto, CA
Founded
2022

Recently Posted Jobs

Explore more opportunities at Hippocratic AI

See more jobs