Inference & Workload Optimization Lead (Horizon AI)

Posted 4 Days Ago
Be an Early Applicant
Hiring Remotely in USA
Remote
Entry level
Cloud • Information Technology • Cybersecurity • Infrastructure as a Service (IaaS)
The Role
Leads GPU performance optimization for customer training and inference workloads. Establishes hardware-specific performance baselines, identifies bottlenecks, tunes distributed workloads, and converts optimization expertise into agent-executable improvements. Partners with customers on workload tuning, while secondarily owning internal model serving, cost- and latency-aware routing, model migration, and GPU allocation decisions.
Summary Generated by Built In
Inference & Workload Optimization Lead

Level: Principal (MTS 5) · Reports to: Chief AI Officer

Horizon AI is building the operating system for how organizations run themselves. The first step is autonomous data centers. A fleet of AI agents senses the state of STN’s GPU infrastructure, reasons about what it finds, and increasingly acts on it. This is running in production against real fleet data, not a prototype. Agents already take action where the risk is bounded and the outcome is verifiable. Everything riskier arrives as a recommendation for a human, and the work is moving decisions from the second category into the first, one class at a time, on evidence rather than optimism.

Data centers are the proving ground, not the destination. We start here because infrastructure is measurable and unforgiving: an agent is either right, or the fleet tells you it was wrong. What we build here generalizes, and we know where it goes next.

The role. Own how well customer AI workloads run on the capacity they have committed to, and build the autonomy that makes that expertise scale past one person’s attention.

This is not a role about serving models to our own agents; that already works. This is about making a client’s training and inference jobs measurably faster on their dedicated cluster, which makes the capacity they bought more valuable.

The problems you inherit

  • Optimization is expert-dependent and manual. There is no systematic profiling of customer workloads and no repeatable playbook.

  • No performance baselines. We cannot yet say what good looks like per hardware generation, which means we cannot distinguish a badly configured customer job from a well configured one.

  • None of it is encoded. Whatever optimization knowledge exists lives in people’s heads rather than in agents that can apply it across every cluster.

  • Internal model serving needs maturing, as a secondary duty. Serving works; routing is basic, with no cost or latency awareness.

  • GPU capacity for internal inference is negotiated manually against a fleet largely committed to customers.

You will

  • Profile customer training and inference workloads and find the real bottlenecks: data loading, collective communication, memory bandwidth, kernel efficiency, batching, parallelism layout

  • Establish performance baselines and a definition of good per hardware generation, starting with current-generation NVIDIA hardware

  • Turn that expertise into agent-executable optimization, so a fix reaches every cluster rather than the one you happened to look at

  • Work with committed-capacity customers, directly or through the operating teams, on workload tuning

  • Own internal model serving as a secondary responsibility, including cost and latency-aware routing and model version migration

  • Advise on GPU allocation between internal inference and customer workloads

You have

  • Deep GPU performance engineering: profiling, memory and bandwidth behaviour, and enough kernel-level understanding to know where time actually goes - Serving stack depth, for example vLLM, SGLang, or TensorRT-LLM, including batching, KV cache management, and quantization trade-offs - Distributed training and inference experience at scale: tensor, pipeline, and data parallelism, and collective communication tuning such as NCCL

  • Fluency with profiling tooling, for example Nsight, DCGM, and framework-level profilers

  • The ability to explain a performance finding to a customer and to encode it for an agent. These are different skills and this role needs both

Nice to have: CUDA kernel work, open-weight model deployment at scale, benchmark design, prior customer-facing technical experience

Skills Required

  • Deep GPU performance engineering experience, including profiling, memory and bandwidth behavior, and kernel-level performance analysis
  • Experience with model serving stacks such as vLLM, SGLang, or TensorRT-LLM
  • Knowledge of batching, KV cache management, and quantization trade-offs
  • Distributed training and inference experience at scale using tensor, pipeline, and data parallelism
  • Experience tuning collective communication, such as NCCL
  • Fluency with profiling tools such as Nsight, DCGM, and framework-level profilers
  • Ability to explain performance findings to customers and encode optimizations for AI agents
  • CUDA kernel development
  • Open-weight model deployment at scale
  • Benchmark design experience
  • Prior customer-facing technical experience
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
99 Employees
Year Founded: 2016

What We Do

STN, Inc. is a managed technology and infrastructure provider serving enterprise, regulated, and AI-driven organizations. It designs, operates, and supports secure, scalable systems, including managed IT, cloud and platform services, cybersecurity, data management, compliance engineering, enterprise hardware and software, and GPU One, its GPU-as-a-Service platform for AI training, tuning, inference, and other high-performance workloads. STN emphasizes reliability, audit readiness, and ongoing human support.

Similar Jobs

Donorbox Logo Donorbox

Head Of Product

Artificial Intelligence • Fintech • Information Technology • Marketing Tech • Payments • Social Impact • Software
Remote
2 Locations
150 Employees
130K-180K Annually

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Inside Sales Representative

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Remote or Hybrid
14 Locations
40000 Employees
45K-85K Annually

Liftoff Logo Liftoff

Tax Manager

AdTech • Artificial Intelligence • Big Data • Machine Learning • Marketing Tech • Mobile • Software
Easy Apply
Remote
United States
660 Employees
124K-175K Annually

Mastercard Logo Mastercard

Manager, Total Rewards - Immigration Strategy

Blockchain • Fintech • Payments • Consulting • Cryptocurrency • Cybersecurity • Quantum Computing
Remote or Hybrid
Harrison, NY, USA
38800 Employees
97K-179K Annually

Similar Companies Hiring

Axle Health Thumbnail
Artificial Intelligence • Healthtech • Information Technology • Logistics
Santa Monica, CA
25 Employees
NODA AI Thumbnail
Artificial Intelligence • Information Technology • Software • Cybersecurity
Austin, TX
54 Employees
Golden Pet Brands Thumbnail
Digital Media • eCommerce • Information Technology • Marketing Tech • Pet • Retail • Social Media
El Segundo, California
178 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account