Palo AltoFull TimeEngineering
Remotely
pythonrustllmdistributed systemsreinforcement learningjaxrl
Job Description
📋 Description 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
Back to all jobs