Software Engineer, RL Training Framework
$180k - $440k • Palo Alto, CA
Posted 1d ago
Job Location
Palo Alto, CA
Tech Stack
Remote Work Policy
On-site
Categories
AI Research Engineer
About the job
The RL infrastructure team is looking for an engineer to help develop our RL training framework. This role involves designing and implementing the systems that support all RL workloads at SpaceXSI, from small-scale experiments to production training runs. You will be responsible for profiling, debugging, and optimizing the end-to-end training performance, as well as improving the scalability and observability of the RL stack.
Responsibilities
- Design and implement systems for all RL workloads, from small scale 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
- Comprehensive medical, vision, and dental coverage
- Access to a 401(k) retirement plan
- Short & long-term disability insurance
- Life insurance
- Various other discounts and perks