Design and implement systems backing RL workloads from ablations to production training runs.
Profile, debug, and optimize end-to-end training performance
Improve scalability and observability of the RL stack

🎯 Requirements

Experience building, debugging, and optimizing efficiency of large-scale distributed systems
Comfortable diving into unfamiliar areas and solving problems at all levels of the stack
Proficiency in Python, Jax, Rust, and/or C++

🎁 Benefits

Equity, medical, vision, and dental coverage
401(k) retirement plan
Short & long-term disability insurance
Life insurance, discounts and perks

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot