Member of Technical Staff - RL Training Framework

Help build and improve the RL training framework by designing and implementing systems that support RL workloads, from small experiments to production training. Profile, debug, and optimize end-to-end training performance, while enhancing scalability and observability across the RL stack. Work in the RL infrastructure team on large-scale distributed systems, diving into unfamiliar areas and improving efficiency using Python and one or more of JAX, Rust, or C++.