S

STN

Vendor-agnostic IT infrastructure and GPU services

Inference & Workload Optimization Lead

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USANo salary listedRemote

About the Job

Requirements

  • Deep GPU performance engineering, including profiling, memory and bandwidth behavior, and kernel-level understanding.
  • 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, including tensor, pipeline, and data parallelism, and collective communication tuning such as NCCL.
  • Fluency with profiling tools such as Nsight, DCGM, and framework-level profilers.
  • Ability to explain performance findings to customers and encode them for an agent.

Responsibilities

  • Profile customer training and inference workloads and identify bottlenecks in data loading, collective communication, memory bandwidth, kernel efficiency, batching, and parallelism layout.
  • Establish performance baselines and define good performance for each hardware generation, starting with current-generation NVIDIA hardware.
  • Turn optimization expertise into agent-executable optimization so fixes can be applied across every cluster.
  • Work with committed-capacity customers, directly or through 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.

Desired Qualifications

  • CUDA kernel work.
  • Open-weight model deployment at scale.
  • Benchmark design.
  • Prior customer-facing technical experience.

About the Company

STN

Vendor-agnostic IT infrastructure and GPU services

STN provides IT infrastructure services, designing, building, operating, and securing technology foundations for enterprise IT and AI workloads. Its offerings include Technology Consulting, Managed IT Services, and Technology Procurement through a VAR model, plus the GPU-One private GPU-as-a-Service for scalable AI and high-performance computing, and STN-One for cybersecurity and risk management. The company differentiates itself by acting as a vendor-agnostic consultant, delivering customized, reliable, and scalable IT environments across sectors such as financial services, healthcare, government, and engineering. Its goal is to help clients achieve business outcomes by providing secure, high-performance infrastructure that supports growth and innovation.

Company size
51-200
Company stage
Not listed
Total funding
Not listed
Headquarters
Not listed
Founded
2016

Recently Posted Jobs

Explore more opportunities at STN

See more jobs