Member of Technical Staff RL Training Framework

You will design and implement systems supporting reinforcement-learning workloads, from small-scale ablations through production training runs. You will profile, debug, and optimize end-to-end training performance while improving the scalability and observability of the reinforcement-learning stack.

Responsibilities

• Design and implement systems supporting reinforcement-learning workloads.
• Profile, debug, and optimize end-to-end training performance.
• Improve the scalability and observability of the reinforcement-learning stack.

Requirements

• Experience building, debugging, and optimizing large-scale distributed systems.
• Proficiency in Python, JAX, Rust, or C++.

Benefits

• Equity
• Medical coverage
• Vision coverage
• Dental coverage
• 401(k) retirement plan
• Short-term disability insurance
• Long-term disability insurance
• Life insurance
• Discounts and perks