> Markdown version of [/jobs/ext/4625868-performance-engineer-inference-training-gpu](https://www.wearedevelopers.com/jobs/ext/4625868-performance-engineer-inference-training-gpu). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Performance Engineer (Inference, Training & GPU)

- **Company:** JOB WORLD
- **Location:** San Francisco, CA, United States
- **Salary:** $200,000.0 - $300,000.0
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, C++ (Programming Language), Nvidia CUDA, Distributed Systems, Fault Tolerance, Python (Programming Language), Graphics Processing Unit (GPU), Pytorch, Large Language Models, Generative AI, Gpu Programming, Discretization, AWS Trainium, Mixed Precision, Stable Diffusion
- **Published:** November 7, 2026
- **Apply:** https://www.juju.com/job/28_25304_e3c7294852f8f13b2f44515bd5c99a30

## About the Role

You should excel at the fundamentals below - we index on inference, serving, GPU optimization, and training performance. Distributed-systems breadth is welcome, but secondary.

 * Strong performance-engineering foundations: profiling, roofline analysis, latency/throughput optimization, and disciplined root-cause investigation.
 * Deep GPU programming and optimization experience (CUDA and/or Triton) - kernel-level tuning, memory hierarchy, and bandwidth optimization at scale.
 * Hands-on experience optimizing inference and serving for large models: batching, KV/prompt caching, quantization, and low-latency, high-throughput sampling.
 * Hands-on experience optimizing training performance: parallelism, distributed communication, mixed/low precision, and utilization.
 * Working knowledge of ML framework internals (PyTorch and/or JAX; torch.compile, XLA, or similar compiler paths).
 * Strong proficiency in Python, with the ability to drop into C++/CUDA (and Rust or Go) as the work demands.
 * High-ownership mindset - you measure yourself by throughput shipped and latency cut, not tickets closed., * Experience at an AI lab or ML-native company, optimizing systems used directly by researchers and productionizing research code.
 * Low-precision and numerics depth: FP8/INT8 quantization, mixed-precision, and detecting numerical regressions across hardware platforms.
 * Distributed systems for large-scale training and inference - collective communication (NCCL), interconnects (NVLink), model and tensor parallelism, and fault tolerance. A strong plus, but not a substitute for the core skills above.
 * Experience serving generative, diffusion, video, or 3D/spatial models - not just text LLMs.
 * Multi-accelerator experience (GPU plus TPU or Trainium) and partnering with hardware vendors on accelerator capabilities.
 * Building performance-modeling and observability frameworks for GPU utilization and cost.

## Description

We are looking for a Performance Engineer to make World Labs' models train and serve as fast as the hardware allows.

Running large generative world models at scale is a novel systems problem. You will find the bottlenecks - in kernels, in the serving path, in the training loop, in how we use our GPUs - and eliminate them. Your ownership is technical and concrete: the throughput you unlock, the latency you cut, the utilization you win back, and the correctness you hold while doing it. You will work up and down the stack, from low-level tensor and kernel optimization to fleet-wide serving efficiency, in close partnership with the researchers whose models you are accelerating., * Optimize inference and serving end to end - latency, throughput, batching, caching, and scheduling - to serve our models efficiently at production scale.

 * Write and tune GPU kernels (CUDA, Triton) for hot paths; drive kernel fusion, memory- and bandwidth-bound optimization, and low-precision (FP8/INT8) execution.
 * Optimize training throughput and GPU utilization: parallelism strategies, communication/compute overlap, mixed precision, and eliminating pipeline stalls.
 * Build performance models, profiling workflows, and observability that make throughput, latency, cost, utilization, and their tradeoffs legible across the stack.
 * Own numerical correctness across precision, kernel, and hardware changes - treating correctness as part of performance, not separate from it.
 * Partner with researchers to productionize models for serving and to make experiments run faster and more reliably.
 * Where needed, work on the distributed systems that training and inference run on - but the core of the job is squeezing the most out of every GPU.