S
Staff Machine Learning Research Engineer, Agent Post-training
Scale AIAI Infrastructure company
San Francisco, United StatesLead
Accel
Y Combinator
Founders Fund
Index Ventures
Coatue
Meta
Data & AI
About the role
TL;DR
Lead the development and optimization of post-training techniques for enterprise-grade generative AI agents.
- •AI is becoming vitally important in every function of our society.
- •At Scale, our mission is to accelerate the development of AI applications.
- •For 9 years, Scale has been the leading AI data foundry, helping fuel the most exciting advancements in AI, including generative AI, defense applications, and autonomous vehicles.
- •With our recent investment from Meta, we are doubling down on building out state of the art post-training algorithms to reach the performance necessary for complex agents in enterprises around the world.
- •The Enterprise ML Research Lab works on the front lines of this AI revolution.
- •We are working on an arsenal of proprietary research, tools, and resources that serve all of our enterprise clients.
- •As a Staff Agent Post-Training MLRE, you will build out our next-gen Agent RL training platform.
- •You’ll build out the platform that will train best-in-class Agents that achieve state of the art results on real enterprise use-cases.
- •You’ll integrate cutting edge research into our training stack, enabling MLREs on the Enterprise AI team to deploy use-cases ranging from next-generation AI cybersecurity firewall LLMs to training foundation healthtech search models.
- •If you are excited about shaping the future of the modern GenAI movement, we would love to hear from you! You will: Train state of the art models, developed both internally and from the community, to deploy to our enterprise customers.
- •Research cutting edge algorithms to integrate directly into our training stack.
- •Design solutions that enable complex multi-agent systems to directly learn from both process + outcome based rewards.
- •Ideally you’d have: 5+ years of LLM training in a production environment Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc.
- •Publications in top conferences such as NEURIPS, ICLR, or ICML within the last two years PhD or Masters in Computer Science or a related field Compensation packages at Scale for eligible roles include base salary, equity, and benefits.
- •The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position and may be inclusive of several career levels at Scale; it will be determined during the interview process based on work location and additional factors, including job-related skills, experience, qualifications, interview performance, and relevant education or training.
- •Scale employees in eligible roles are also granted equity based compensation, subject to Board of Director approval.
- •Your recruiter can share more about the specific salary range for your preferred location during the hiring process, and confirm whether the hired role will be eligible for equity grant.
- •You'll also receive benefits including, but not limited to: comprehensive health, dental and vision coverage, retirement benefits, a learning and development stipend, and generous PTO.
- •Additionally, this role may be eligible for additional benefits such as a commuter stipend.
- •Please reference the job posting's subtitle for where this position will be located.
- •For pay transparency purposes, the base salary range for this full-time position in the locations of San Francisco, New York, Seattle is: $250,000-$350,000 USD PLEASE NOTE: Our policy requires a 90-day waiting period before reconsidering candidates for the same role.
- •This allows us to ensure a fair and thorough evaluation of all applicants.
- •About Us: At Scale, our mission is to develop reliable AI systems for the world's most important decisions.
- •Our products provide the high-quality data and full-stack technologies that power the world's leading models, and help enterprises and governments build, deploy, and oversee AI applications that deliver real impact.
- •We work closely with industry leaders like Meta, Cisco, DLA Piper, Mayo Clinic, Time Inc., the Government of Qatar, and U.S. government agencies including the Army and Air Force.
- •We are expanding our team to accelerate the development of AI applications.
- •We believe that everyone should be able to bring their whole selves to work, which is why we are proud to be an inclusive and equal opportunity workplace.
- •We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability status, gender identity or Veteran status.
- •We are committed to working with and providing reasonable accommodations to applicants with physical and mental disabilities.
- •If you need assistance and/or a reasonable accommodation in the application or recruiting process due to a disability, please contact us at [email protected].
- •Please see the United States Department of Labor's Know Your Rights poster for additional information.
- •We comply with the United States Department of Labor's Pay Transparency provision.
- •PLEASE NOTE: We collect, retain and use personal data for our professional business purposes, including notifying you of job opportunities that may be of interest and sharing with our affiliates.
- •We limit the personal data we collect to that which we believe is appropriate and necessary to manage applicants’ needs, provide our services, and comply with applicable laws.
- •Any information we collect in connection with your application will be treated in accordance with our internal policies and programs designed to protect personal data.
- •Please see our privacy policy for additional information.
Required skills
PythonPyTorchLLMsNLP
Domain expertise
ai
Tech stack
PythonPyTorchLLMsNLP