Model Engineer, Rl Training Framework
xAI- Location
- Palo Alto, California
- Work mode
- Remote
- Type
- full time
- Listed
Applications go directly to xAI. Ramai takes no fee for placement and no money for ranking.
[ About this role ]
Member of Technical Staff - RL Training Framework at Cabana in Palo Alto, California. This role involves designing and implementing systems for RL workloads, profiling and optimizing training performance, and improving the scalability and observability of the RL stack. Requires experience with large-scale distributed systems and proficiency in Python, Jax, Rust, and/or C++.
Skills and focus areas
- rl training framework development
- large-scale distributed systems
- performance profiling
- debugging
- optimization
- scalability
- observability
- python
- jax
- rust
- c++
- llm training infrastructure
Summary written by Ramai from the original posting. Full details and requirements are on xAI’s careers page.
[ More at xAI ]
See all 117 open roles at xAI →[ Similar roles at other startups ]
Browse all open engineering roles →[ Not quite right? ]
Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.