Member of Technical Staff

We're looking for a Member of Technical Staff with 0-6 YOE to join our team building AI performance engineers — agents that automatically optimize GPU kernels and ML workloads to run as efficiently as possible on any hardware. You'll work at the intersection of AI agent systems and low-level GPU engineering, helping expand support to new hardware architectures and fulfilling growing demand from chip companies and cloud providers. This is a high-ownership, zero-bureaucracy role where you're in direct contact with customers from week one.ResponsibilitiesBuild and improve the agent framework for GPU kernel optimization — multi-turn tool use, state management, reward signalsWrite and debug GPU kernels and low-level compiler code to understand and improve agent outputDevelop integrations with GPU profilers and compiler toolchains across CUDA, HIP, and other hardware targetsDesign architecture for remote GPU execution across cloud providersWork directly with chip company and cloud provider stakeholders (AMD, AWS, DigitalOcean, and others) as a forward-deployed engineerDrive long-term research bets and push the technical frontier for new hardware architectures