Preference Model
Member of Technical Staff - Machine Learning Capabilities, New Graduates
$165K–$300K
San Francisco · On-site · Staff · Full-time · Posted May 4, 2026
- Visa sponsorship
- Visa sponsorship available.
About the job
Design and build reinforcement learning environments that train frontier models on machine-learning research and engineering tasks.
What you'll do
- Develop RL environments and reward schemes, and conduct experiments and evaluations.
- Deliver environments into production training runs and collaborate on research and engineering.
- Develop expertise in frontier ML research, training, and inference infrastructure.
What they ask for
- Strong machine-learning fundamentals and expert knowledge in an active DL/ML research area, with publications or public code.
- Deep understanding of transformer internals.
- Proficiency in Python, Numpy, and systems programming.
- Recent graduate able to start soon.
Nice to have
- PhD research experience and publications.
- Kernel development expertise, RL environment work, open-source ML infrastructure contributions, or cloud/infrastructure-as-code experience.
Pay
$165K – $200K USD base salary; $0 to $300K+ USD bonus range based on performance
Tools
Python, Numpy, PyTorch, JAX, CUDA, Triton, Pallas, AWS
AI focus
Build reinforcement learning environments to advance frontier model capabilities in ML research and engineering.
Model-written summary · the original posting is authoritative