Research Engineer, Performance RL

We're hiring a Research Engineer for our Code RL team to advance our models' ability to safely write correct, fast code for accelerators. The ideal candidate will have expertise in accelerators, ML framework programming, and experience working across the stack.

Requirements

• Expertise with accelerators (CUDA, ROCm, Triton, Pallas)
• ML framework programming (JAX or PyTorch)
• Experience working across the stack – kernels, model code, distributed systems
• Ability to balance research exploration with engineering implementation

Benefits

• Competitive compensation
• Benefits
• Optional equity donation matching
• Generous vacation and parental leave
• Flexible working hours