Member of Technical Staff - Machine Learning Capabilities

About the role

We are seeking experienced Machine Learning Engineers to design and build reinforcement learning environments to safely advance model capabilities in machine learning research and engineering. Your primary responsibility will be to teach frontier models the work of an ML engineer or researcher at a frontier lab, blending research and engineering.

Responsibilities

• Design and build RL environments and reward functions that produce clean, learnable signals for frontier models on ML research and engineering tasks
• Collaborate with others to brainstorm and create new ideas and tools to improve the environment building process

Requirements

• You have strong ML fundamentals and broad research interests
• You read many papers or tutorials, understand topics deeply, and have the creativity to translate them into RL problems
• Proficiency in Python and systems programming, and at least one of PyTorch or JAX
• Problem solvers who take ownership and drive solutions end-to-end
• Possess passion for staying current with the rapidly evolving ML infrastructure landscape
• Able to meet throughput expectations and respond quickly to feedback

Qualifications

• Expert knowledge in an active DL/ML research area, with publications or public code to show for it (PhD, MS preferred)
• Deep understanding of transformer internals, training/inference of modern LLMs, experience with inference libraries (vLLM, SGLang, etc)
• Strong expertise in kernel development (CUDA, Triton, Pallas)
• Built complex interactive RL environments

Skills

• Strong ML fundamentals and broad research interests
• Python and systems programming proficiency, and at least one of PyTorch or JAX
• Expertise in an active DL/ML research area, with publications or public code to show for it
• Deep understanding of transformer internals, training/inference of modern LLMs, experience with inference libraries
• Strong expertise in kernel development (CUDA, Triton, Pallas)
• Built complex interactive RL environments

Benefits

• Competitive cash and equity compensation
• Ownership and autonomy in a fast-moving startup environment
• Opportunity to work with top machine learning engineers
• Health, vision, dental, benefits
• 401K match
• Lunch provided everyday onsite
• Weekly snack orders
• Visa sponsorship & relocation support available

Pay

$200K - $350K

Schedule

Not specified