Sr. Model Engineer, RL Training Framework
xAI · Palo Alto, CA
Frontier labonsitesenior$180k–$400k
At a glance
Highlights
- Design systems backing all RL workloads from ablations to production training.
- Help build Super Intelligence to help humanity understand the universe.
- Profile and optimize end-to-end training performance at massive scale.
What they offer
- equity
- comprehensive medical, vision, and dental coverage
- 401(k) retirement plan
- short & long-term disability insurance
- life insurance
Ideal candidate
A systems-minded builder fascinated by large-scale learning who loves debugging performance and scaling RL infrastructure from small ablations to production.
Requirements
- Experience writing production-quality code in Python, PyTorch, JAX, Rust, or C++
- Experience training large models (e.g., pre-training, post-training, fine-tuning, or RL training of neural networks with 1B+ parameters)
Lab profile
About xAI
Builds AI systems to understand the universe and advance human knowledge.
Known for
- Engineering excellence focus
- Large-scale data center operations
Posted · Last seen