Member of Technical Staff - Machine Learning Capabilities, New Graduates
Core
Design and build reinforcement learning (RL) environments and reward schemes to teach frontier models to perform machine learning research and engineering tasks.
Role type
New graduate machine learning engineer (RL environments)
Builds
RL training environments and reward functions for frontier LLMs
Domain
Artificial Intelligence / Machine Learning Research / Reinforcement Learning
Deliverable
production ML models
Required skills
ML fundamentals, deep learning research, transformer internals, Python, Numpy, systems programming, PyTorch or JAX
Preferred skills
CUDA/Triton/Pallas kernel development, cloud platform experience, infrastructure-as-code
Technologies
PyTorch, JAX, Python, Numpy, CUDA, Triton, Pallas
Responsibilities
Design and implement RL environments, conduct experiments and evaluations, deliver work into production training runs, collaborate on environment building tools
Seniority
Junior, new graduate