Member of Engineering – Reinforcement Learning

Posted 2hrs ago

Employment Information

Education
Salary
Experience
Job Type

Report this job

Job expired or something wrong with this job?

Job Description

Member of Engineering developing reinforcement learning to enhance reasoning and coding capabilities for language models. Join a remote-first team pushing the frontier of AI advancements.

Responsibilities:

  • Research and experiment on ways to improve reasoning and code generation for LLMs. Own the full experiment life cycle from idea to experimentation and integration
  • Keep up with the latest research, and be familiar with the state of the art in LLMs, RL, and code generation. Translate research ideas into clean, reusable codebases that other researchers can build on
  • Design, analyze, and iterate on data generation and training of LLMs
  • Implement and iterate on RL training pipelines that scale reliably across domains
  • Diagnose training instabilities and failures, debug RL runs and propose mitigation methods
  • Write high-quality, reproducible and maintainable code

Requirements:

  • Experience with Large Language Models (LLM), including:
  • Understanding of the Transformer architecture and scaling laws
  • Mid-training and post-training techniques
  • Experience training reasoning and/or agentic models
  • Hands-on use of LLMs, with a sense of their capabilities and limitations
  • Reinforcement Learning experience
  • Solid grasp of Reinforcement Learning concepts and familiarity with modern algorithms
  • Experience developing distributed, large-scale RL pipelines from data creation to evaluations
  • Research experience
  • Scientific publications in any of the following topics: Reinforcement Learning, LLMs and reasoning models
  • Ability to discuss the latest research with sufficient level of detail
  • Is reasonably opinionated
  • Engineering skills
  • Strong machine learning, algorithm skills and engineering background
  • Experience with distributed training
  • Excellent programming skills in Python
  • Familiarity with a deep learning framework (Pytorch or JAX)

Benefits:

  • Fully remote work & flexible hours
  • 37 days/year of vacation & holidays
  • Health insurance allowance for you and dependents
  • Company-provided equipment
  • Wellbeing, always-be-learning and home office allowances
  • Frequent team get togethers
  • Great diverse & inclusive people-first culture