SDE IV - GPU Engineer

About the role

As a GPU Systems Engineer , you’ll lead design and optimization efforts across our GPU inference stack. You will architect the libraries and runtime systems that enable Stable Diffusion, multimodal transformers , and emerging video generation models to run efficiently at scale. You’ll guide cross-functional teams, influence hardware selection, and set the technical vision for GPU optimization practices across the company.

What you'll do

• Architect high-performance inference runtimes , kernel dispatchers, and memory planners for large diffusion and transformer workloads
• Lead investigations into cross-GPU performance bottlenecks , communication overheads, and scheduling inefficiencies
• Drive multi-GPU parallelism strategies — model, pipeline, and tensor parallelization
• Establish company-wide GPU optimization standards, tooling, and SLIs
• Collaborate with research to design scalable implementations of novel architectures

What you need

• C++
• Machine learning
• LLMs
• Any graduate