Member of Technical Staff Research Engineer (GPU Performance)

We are building AI to simulate the world through merging art and science.

We believe that world models are at the frontier of progress in artificial intelligence.

Language models alone won’t solve the world’s hardest problems – robotics disease scientific discovery.

Real progress requires models that experience the world and learn from their mistakes the same way that humans do.

And this kind of trial and error can be massively accelerated when done in simulation rather than in the real world.

World models offer the most clear path to general-purpose simulation changing how stories are told how scientific progress is made and how the next frontiers of humanity are reached.

Our team consists of creative open minded caring and ambitious people who are determined to change the world.

We aspire to continuously build impossible things and our ability to do so relies on building an incredible team.

If you are driven to do the same we'd love to hear from you.

ABOUT THE ROLE We’re looking for Research Engineers to help our world models train faster and run more efficiently without compromising what they can do.

You will profile optimize and rearchitect the systems that turn research ideas into models that run at scale and in real time — directly shaping what is computationally possible and by extension what capabilities we can build.

WHAT YOU’LL DO - Optimize training throughput across large GPU clusters — improving MFU through custom kernels mixed-precision strategies (FP8 BF16) memory-efficient attention and activation checkpointing - Design and maintain distributed training infrastructure: tensor parallelism context parallelism FSDP and fault-tolerant multi-node setups - Profile and accelerate inference pipelines for real-time multimodal generation — CUDA graph compilation KV cache optimization operator fusion and latency reduction - Optimize and scale our training infrastructure to improve efficiency and reliability - Contribute to the entire stack from low-level kernel optimizations to high-level model design WHAT YOU’LL NEED - 4+ years of experience in systems engineering ML infrastructure or performance optimization for deep learning - Familiarity with GPU kernel development (CUDA Triton CUTLASS) and distributed systems (NCCL collective communication model parallelism) - Experience with ML framework internals (PyTorch JAX) and mixed-precision / low-precision techniques (FP8 INT8) - Experience building and operating large-scale training infrastructure including fault tolerance and cluster orchestration - Excitement about building AI that simulates the world — and making it performant enough to run in real time - Bonus if you have experience with torch’s compilation feature Runway strives to recruit and retain exceptional talent from diverse backgrounds while ensuring pay equity for our team.

Our salary ranges are based on competitive market rates for our size stage and industry and salary is just one part of the overall compensation package we provide.

There are many factors that go into salary determinations including relevant experience skill level and qualifications assessed during the interview process and maintaining internal equity with peers on the team.

The range shared below is a general expectation for the function as posted but we are also open to considering candidates who may be more or less experienced than outlined in the job description.

In this case we will communicate any updates in the expected salary range.

Lastly the provided range is the expected salary for candidates in the U.S.

Outside of those regions there may be a change in the range which again will be communicated to candidates.

WORKING AT RUNWAY Great things come from great teams. https://www.youtube.com/watch?v=kwmj4ato2kw&ab_channel=Runway We’d love to hear from you.

We’re committed to creating a space where our employees can bring their full selves to work and have equal opportunity to succeed.

So regardless of race gender identity or expression sexual orientation religion origin ability age veteran status if joining this mission speaks to you we encourage you to apply.

More about Runway - Universal World Simulator https://runwayml.com/world-simulator.html - GWM-1 https://runwayml.com/research/introducing-runway-gwm-1 - Gen-4.5 https://runwayml.com/research/introducing-runway-gen-4.5 - General World Models https://runwayml.com/research/introducing-general-world-models - Robotics SDK https://runwayml.com/research/introducing-runway-gwm-1#robotics-section - Conversational Real-time Agents https://runwayml.com/research/introducing-runway-gwm-1#avatars-section - Runway Studios https://runwayml.com/studios We're excited to be recognized as a best place to work: Crain's https://www.crainsnewyork.com/awards/best-places-work-2023 | InHerSight https://www.inhersight.com/companies/best/city/new-york-city-ny | BuiltIn NYC https://builtin.com/awards/new-york-city/2024/best-places-to-work | INC https://www.inc.com/best-workplaces/2024