Member of Technical Staff - Mid-Training Infra
Reflection AI · San Francisco, California
- Location
- San Francisco, California, United States
- Employment
- Full time
- Level
- Staff
- Posted
- 5 months ago
About this role
Our Mission
Reflection is a research lab making intelligence open and accessible for everyone to use, customize, and build on. We build open models that let anyone control their intelligence and help shape the future of AI. Our mission: make intelligence open and accessible to all.
About the Role
Design, build, and operate large-scale GPU infrastructure for high-throughput model inference and mid-training workloads.
Develop systems that power synthetic data generation and reinforcement learning pipelines at scale.
Build high-performance inference platforms capable of serving and evaluating models across thousands of GPUs.
Optimize throughput, latency, and GPU utilization for large language model inference and rollout workloads.
Build infrastructure that supports reinforcement learning pipelines, including large-scale rollout generation, evaluation, and policy improvement loops.
Work closely with research teams to support distributed RL workloads and large-scale model evaluation infrastructure.
Improve performance of model execution through kernel-level optimization, model parallelism strategies, and GPU runtime improvements.
Develop distributed systems that enable large-scale synthetic data generation and RL-driven training workflows.
Diagnose and resolve performance bottlenecks across inference runtimes, GPU kernels, networking, and distributed compute systems.
Ideal Experience
Experience deploying and operating large-scale GPU systems for inference or model serving.
Several years of hands-on experience building and running production infrastructure.
Strong understanding of GPU performance characteristics and optimization techniques.
Experience working with modern inference frameworks such as SGLang, Megatron, or similar high-performance LLM runtimes.
Familiarity with distributed reinforcement learning infrastructure or rollout generation systems.
Experience optimizing throughput for large-scale model execution workloads.
Experience working with GPU kernels or low-level performance optimization.
Familiarity with infrastructure used for synthetic data pipelines or RL training workflows.
Experience debugging performance issues across GPU, networking, and distributed execution layers.
What We Offer:
We believe that to make intelligence open and accessible to all, you need to start at the foundation. Joining Reflection means building from the ground up as part of a talent-dense team. You will help define our future as a company, and help define the future of open foundational models.
We want you to do the most impactful work of your career with the confidence that you and the people you care about most are supported.
Top-tier compensation: Salary and equity structured to recognize and retain our talent globally.
Stock options: Everyone who joins and contributes to Reflection's success gets to share in the upside through stock options.
Health & wellness: Comprehensive medical, dental, vision, and life, with an annual wellness allowance.
Meals: Lunch and dinner are provided in the office daily.
Life & family: 22 weeks paid parental leave for all new birthing and non-birthing parents, including adoptive and surrogate journeys.
Vacation days: Unlimited paid time off in the U.S. and 30 days in the U.K.
Sponsorship support: We sponsor visas to help exceptional talent join our team and support long-term immigration pathways where applicable.
Team building: We have regular off-sites, happy hours, and team celebrations.
Export Control Notice: This position may require access to technology or source code subject to the U.S. Export Administration Regulations. Any offer of employment for this role may be conditioned on the Company's ability to provide the candidate with access to such technology or source code in compliance with applicable U.S. export control laws, which may require the Company to seek government authorization.
As published by Reflection AI. Applications are handled on their site.
Skills this posting mentions
About Reflection AI
We are a team of world-class AI researchers building superhuman general agents that automate knowledge work done on a computer.
All 61 openings at Reflection AIOne click, then it is written
Apply to Reflection AI with a resume written for this role.
Queue Member of Technical Staff - Mid-Training Infra and I read the posting, rewrite your resume against it, draft the cover letter, and score the fit. Then you press send, or press one button and I fill in Reflection AI’s form for you.
01Drop your resume
A PDF or a LinkedIn URL. About a minute, once.
02I rank the openings
Every weekday morning, the live catalog scored against your history. Up to 20 worth your time, not two hundred links.
03Each one is written up
Resume rewritten for the posting, a cover letter, a fit score. Press send, or let me fill in the form.
- 25 sent a week, free
- No card
- Nothing sent until you say so
More roles at Reflection AI
See all- 6 weeks ago
Subject Matter Expert - Child Safety
New York City, New York
- 6 weeks ago
Subject Matter Expert - Election Integrity, Violent Extremism, and Fraud
New York City, New York
$80k - $250k/yr - 7 weeks ago
- 7 weeks ago
- 7 weeks ago
- 7 weeks ago
Member of Technical Staff - Research Software Engineer - Safety Evaluations Infrastructure
New York City, New York
$65k - $350k/yrStaffEngineering
Similar roles elsewhere
See more- Today
Support Engineer, AI Infrastructure & Tooling
FigmaSan Francisco, CA • New York
$169k - $245k/yrMid levelEngineering
Put this to work
Paste your career in once. Every application after that is written for you.
Drop a resume or a LinkedIn URL. I rank the live openings against it, rewrite the resume and write a cover letter for the best of them, and fill in the employer's form when you press the button. You read, you decide what goes out.
01Drop your resume
A PDF or a LinkedIn URL. About a minute, once.
02I rank the openings
Every weekday morning, the live catalog scored against your history. Up to 20 worth your time, not two hundred links.
03Each one is written up
Resume rewritten for the posting, a cover letter, a fit score. Press send, or let me fill in the form.
- New matches ranked and written before you are up.
- Every bullet stays inside what your history supports. Nothing invented.
- Queued, submitted, interviewing, offer: one screen, not a spreadsheet.
500 free credits on sign-up. No card. Nothing is sent until you say so.
Listed from the job board Reflection AI publishes. Refolk is not the employer and does not handle their hiring. Applications go to Reflection AI directly.