STN Inc Logo

STN Inc

Inference & Workload Optimization Lead (Horizon AI)

Posted Yesterday
Be an Early Applicant
Remote
Hiring Remotely in USA
Entry level
Remote
Hiring Remotely in USA
Entry level
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.
The summary above was generated by AI
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

Similar Jobs

33 Minutes Ago
Easy Apply
Remote or Hybrid
3 Locations
Easy Apply
105K-150K Annually
Junior
105K-150K Annually
Junior
Cloud • Information Technology • Security • Software • Cybersecurity
Create technical product presentations, gather customer requirements, lead product evaluations and configurations, and design test plans to guide customers toward successful solutions. The role requires cybersecurity and networking expertise, including TCP/IP, DNS, routing, VPNs, and firewalls, along with familiarity with AI tools and AI/ML solutions. This remote Sales Engineer position supports prospective customers and collaborates with sales and technical teams.
Top Skills: AIAi/MlCybersecurityDnsFirewallsGenerative AiLlmsNetworkingRoutingTcp/IpVpn
An Hour Ago
Easy Apply
Remote
United States
Easy Apply
150K-225K Annually
Expert/Leader
150K-225K Annually
Expert/Leader
Artificial Intelligence • Consumer Web • Digital Media • Fintech • Marketing Tech • Software • Financial Services
Build and lead Bankrate’s first-line risk and controls function, including risk committee governance, compliance monitoring, mitigation roadmaps, policy management, third-party due diligence, certifications, and employee training. The role partners with Legal, Compliance, Security, Procurement, and Data Privacy while using AI-enabled workflows to scale governance activities. It requires financial services experience, strong mortgage knowledge preferred, and the ability to manage complex cross-functional initiatives from scratch.
Top Skills: Agentic WorkflowsAi ToolsIsoSocWcag
An Hour Ago
Easy Apply
Remote
United States
Easy Apply
115K-194K Annually
Mid level
115K-194K Annually
Mid level
Cloud • Security • Software • Cybersecurity • Automation
Designs, operates, and improves production Kubernetes platforms across cloud environments. Manages ArgoCD GitOps workflows, Terraform infrastructure, CI/CD integrations, platform reliability, observability, security controls, and developer enablement. Collaborates with application teams to onboard services, create self-service tooling and documentation, respond to incidents, and improve platform availability and performance.
Top Skills: ArgocdAWSAzureBashCiliumCrossplaneDnsExternal Secrets OperatorGCPGithub ActionsGitlab CiGitopsGoGrafanaHelmIngressIstioKubernetesKyvernoLinkerdLoad BalancingLokiOpa/GatekeeperOpentelemetryPrometheusPythonSealed SecretsService MeshTektonTerraformVault

What you need to know about the Boston Tech Scene

Boston is a powerhouse for technology innovation thanks to world-class research universities like MIT and Harvard and a robust pipeline of venture capital investment. Host to the first telephone call and one of the first general-purpose computers ever put into use, Boston is now a hub for biotechnology, robotics and artificial intelligence — though it’s also home to several B2B software giants. So it’s no surprise that the city consistently ranks among the greatest startup ecosystems in the world.

Key Facts About Boston Tech

  • Number of Tech Workers: 269,000; 9.4% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Thermo Fisher Scientific, Toast, Klaviyo, HubSpot, DraftKings
  • Key Industries: Artificial intelligence, biotechnology, robotics, software, aerospace
  • Funding Landscape: $15.7 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Summit Partners, Volition Capital, Bain Capital Ventures, MassVentures, Highland Capital Partners
  • Research Centers and Universities: MIT, Harvard University, Boston College, Tufts University, Boston University, Northeastern University, Smithsonian Astrophysical Observatory, National Bureau of Economic Research, Broad Institute, Lowell Center for Space Science & Technology, National Emerging Infectious Diseases Laboratories

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account