Sr. Model Engineer, RL Inference
xAI, Palo Alto, CA
The role
The RL infrastructure team is looking for an engineer to help with low precision RL training and inference.
What you will do
- Design and optimize our inference stack for all shapes of RL workloads at SpaceXSI, from small scale ablations to production training runs
- Analyze, profile and address performance bottlenecks in large scale RL systems
- Work closely with the modeling team to efficiently implement novel RL techniques and algorithms