Skip to Content

Job Overview

As a Senior Staff Machine Learning Performance Engineer, you will own that speed: working with the modeling teams on their hardest bottlenecks, setting the technical direction for performance, and turning one-off wins into defaults — including the agentic…

Key points

  1. Own step time, scaling efficiency and long-run stability for foundation-model training across thousands of accelerators — from the data path to kernels…

  2. Cross-Team Technical Leadership: Work directly with the modeling, infrastructure, simulation, data-mining and robotics teams on their hardest bottlenecks…

  3. Turn one-off wins into modular components that become defaults across most models and workloads, rather than a patch that helps a single job.

  4. Extend performance tooling and metric coverage across the stack, while keeping instrumentation overhead low enough to stay on for production jobs fleet-wide.

  5. Hands-on experience optimizing GPU training or inference workloads in production, at multi-node scale.

  6. A track record of diagnosing performance problems in real production workloads and landing the fixes — often in systems owned by other teams —…

  7. In-depth understanding of computer architecture and high-performance computing.

  8. Strong C++ and Python, with a solid command of data structures and algorithms.

Skills

Machine LearningNode.jsPyTorchPythonLLMsCommunication

Experience: Senior-level · 8+ years

Similar jobs

More open roles like this one