Uniphore Logo

Uniphore

Staff Inference Engineer

Posted 18 Days Ago
Be an Early Applicant
In-Office
Palo Alto, CA, USA
175K-219K Annually
Senior level
In-Office
Palo Alto, CA, USA
175K-219K Annually
Senior level
Build and optimize a context-aware inference runtime for fine-tuned small language models across cloud, on-premises, NVIDIA, and AMD environments. Responsibilities include KV-cache management, speculative decoding, GPU kernel optimization, continuous batching, quantization, multi-GPU execution, adapter lifecycle management, observability, tenant isolation, reliability, and safe deployment. The role partners with research, model engineering, agent, knowledge systems, and platform teams.
The summary above was generated by AI

Uniphore is the Business AI company. Our sovereign, composable and secure AI platform connects enterprise data, fine-tunes AI models and deploys agentic AI across the enterprise. We empower every worker to boost productivity and help businesses grow faster, operate smarter and reduce costs. Trusted by more than 2,000 businesses globally, and recognized on the Deloitte Fast 500, Uniphore delivers on the promise of AI as a transformative force for business.


Job Description:
 

Staff Inference Engineer –

Uniphore is building an enterprise AI pla4orm where agents operate across voice, video, text, and structured data in real time. We are looking for a Senior or Staff Inference Engineer to own the inference layer — combining deep model inference expertise with systems engineering to deliver a context-aware runtime serving finetuned SLMs across cloud, on-premises, and widely used GPU platforms.


This is a hands-on individual contributor role working closely with Uniphore’s research, model engineering, knowledge systems, agent, and pla4orm teams.

What You Will Build

Context-aware runtime: Integrate context selection, compression, conversation compaction, and persistent task state with model execution. Adapt context budgets, model selection, and reasoning effort to each task.

KV-cache and computation optimization: Build paged KV-cache and prompt-prefix-cache management, cache-aware routing, retention, eviction, and offloading. Preserve correctness as context, permissions, model versions, and adapters change.

Inference performance: Implement and tune speculative decoding, attention kernels, continuous batching, quantization, chunked prefill, and multi-GPU parallel execution.

SLM and adapter lifecycle: Build model and adapter lifecycle management, multi-adapter serving, zero-downtime hot swapping, and resource scheduling.

Production reliability: Implement observability (DCGM, Prometheus, Grafana), tenant isolation, overload protection, failure recovery, and safe rollout and rollback.

Cross-environment deployment: Build portable runtime integrations across NVIDIA and AMD GPU pla4orms, cloud and on-premises environments, and single- or multi-GPU configurations.


Technical Expertise

Model inference internals: Autoregressive generation, prefill vs. decoding, attention architectures, compute- vs. memory-bound execution, numerical precision.

Speculative decoding: Draft-and-verify execution, acceptance/rejection sampling, adaptive speculation, and throughput tradeoffs.

GPU kernels: Flash attention-style kernels, tiling, operator fusion, memory access patterns.

CUDA, HIP, Triton, or CUTLASS proficiency.

KV and prefix caching: Paged KV storage, fragmentation, cache quantization, offloading, cross request prefix reuse, tenant isolation.

Context-window management: Token budgeting, context selection, prompt compression, conversation compaction, sliding window and sparse attention.

Serving and scaling: Continuous batching, chunked prefill, admission control, quantization, multi-GPU parallelism, prefill/decode disaggregation.


Qualifications Required:

• 5+ years engineering, including 3+ years hands-on with GPU systems, distributed inference, or ML serving

• Production experience with vLLM, SGLang, or TensorRT-LLM

• Strong Python and C++ (C++17/20); ability to read and debug CUDA C++ or HIP kernels

• Hands-on GPU profiling and performance optimization

 

Preferred:

• Custom GPU kernel development (CUDA, HIP, Triton, CUTLASS)

• PyTorch internals — torch. Compile, custom operators, MoE or disaggregated

prefill/decode

• LoRA/SLM adapter serving and zero-downtime hot swapping

• NVIDIA and AMD Instinct hardware in production

• Telemetry with DCGM, AMD SMI, Prometheus, and Grafana



 


Hiring Pay Range:

$175,000 - $218,700

Benefits:


In addition to competitive base pay, this position also includes an annual incentive opportunity based on target achievement,  pre-IPO stock options, benefits including medical, dental, vision, 401(k) with a match, and more, plus generous paid time off, paid holidays, paid day off for your birthday and other paid leave policies to support employees through all phases of life.


Location preference:

USA - CA - Palo Alto

Uniphore is an equal opportunity employer committed to diversity in the workplace. We evaluate qualified applicants without regard to race, color, religion, sex, sexual orientation, disability, veteran status, and other protected characteristics.
 
For more information on how Uniphore uses AI to unify—and humanize—every enterprise experience, please visit www.uniphore.com.

HQ

Uniphore Palo Alto, California, USA Office

1001 Page Mill Road, Palo Alto, CA, United States

Similar Jobs

12 Days Ago
In-Office
San Francisco, CA, USA
250K-320K Annually
Senior level
250K-320K Annually
Senior level
Artificial Intelligence • Information Technology
Own production reliability for a Kubernetes-native control plane provisioning and operating GPU inference fleets. Responsibilities include managing host lifecycle state machines, troubleshooting GPU, CUDA, InfiniBand, RDMA, and model-serving failures, extending automated remediation, owning SLOs and error budgets, leading incident response, improving inference performance and efficiency, and partnering across infrastructure, ML platform, research, and product teams.
Top Skills: CudaCustom Resource Definitions (Crds)Event-Driven ArchitecturesGpusInfinibandKubernetesKubernetes ControllersLogsMetricsObservabilityRdmaTraces
18 Days Ago
Remote or Hybrid
Santa Clara, CA, USA
175K-265K Annually
Senior level
175K-265K Annually
Senior level
Artificial Intelligence • Machine Learning • Software
Own end-to-end performance of LLM inference workloads across orchestration, networking, serving frameworks, runtimes, operating systems, and accelerators. Diagnose regressions through profiling, improve latency, throughput, and stability, encode workload requirements into orchestration, develop telemetry and benchmarks, create investigation playbooks, and mentor engineers. Work spans Kubernetes, heterogeneous hardware, speculative decoding, KV-cache management, batching, NUMA, and cross-team root-cause resolution.
Top Skills: AutoscalingBenchmarkingCC++Cpu PinningCpusetsCudaGpuKubernetesKv-CacheLinuxLlm Inference ServingNcclNetworkingNsightNumaPciePythonQuantizationRustSglangSpeculative DecodingTelemetryTensorrt-LlmVllm
18 Days Ago
Hybrid
Santa Clara, CA, USA
155K-235K Annually
Senior level
155K-235K Annually
Senior level
Artificial Intelligence • Machine Learning • Software
Own end-to-end testing and release engineering for an inference cloud platform. Build automated tests and infrastructure for Kubernetes deployments across GPU, Corsair, on-premises, and Azure environments. Define release strategies, versioning, promotion, rollback, compatibility testing, and release gates covering functionality, accuracy, upgrades, and performance. Partner across engineering teams to improve test quality, enable feature-owned testing, and scale reliable releases for distributed AI serving systems.
Top Skills: AksArgoAzureAzure DevopsAzure NetworkingBuildkiteChainsawCorsair AcceleratorsEnvtestGithub ActionsGitlab CiGpuHelmKubernetesKuttlPytestPythonRustSglangTensorrt-LlmVllm

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account