Senior Reinforcement Learning Engineer

poolside • United Kingdom
Remote
Apply
AI Summary

Poolside is seeking a Senior Reinforcement Learning Engineer to improve reasoning and coding abilities of Large Language Models through reinforcement learning. The ideal candidate will have experience with Large Language Models, Reinforcement Learning, and distributed training. They will work on designing and scaling up RL environments, implementing RL training pipelines, and debugging training instabilities.

Key Highlights
Improve reasoning and coding abilities of Large Language Models
Design and scale up RL environments
Implement RL training pipelines and debug training instabilities
Key Responsibilities
Research and experiment on ways to improve reasoning and code generation for LLMs
Design, analyze, and iterate on data generation and training of LLMs
Implement and iterate on RL training pipelines that scale reliably across domains
Technical Skills Required
Large Language Models Reinforcement Learning Distributed Training Transformer Architecture PyTorch JAX
Benefits & Perks
Fully remote work
Flexible hours
37 days/year of vacation & holidays
Health insurance allowance
Company-provided equipment
Wellbeing, always-be-learning and home office allowances
Nice to Have
Scientific publications in Reinforcement Learning, LLMs and reasoning models

Job Description


About Poolside

In this decade, the world will create Artificial General Intelligence. There will only be a small number of companies who will achieve this. Their ability to stack advantages and pull ahead will define the winners. These companies will move faster than anyone else. They will attract the world's most capable talent. They will be on the forefront of applied research, engineering, infrastructure and deployment at scale. They will continue to scale their training to larger & more capable models. They will be given the right to raise large amounts of capital along their journey to enable this. They will create powerful economic engines. They will obsess over the success of their users and customers.

poolside exists to be this company - to build a world where AI will be the engine behind economically valuable work and scientific progress.

View GDPR Policy

About Our Team

We are a remote-first team that sits across Europe and North America and comes together once a month in-person for 3 days and for longer offsites twice a year.

Our R&D and production teams are a combination of more research and more engineering-oriented profiles, however, everyone deeply cares about the quality of the systems we build and has a strong underlying knowledge of software development. We believe that good engineering leads to faster development iterations, which allows us to compound our efforts.

About The Role

You would be working on our reinforcement learning team focused on improving reasoning and coding abilities of Large Language Models through reinforcement learning. This is a hands-on role where you’ll work end-to-end from researching new exploration or training algorithms, to designing and scaling up RL environments, to implementing your ideas across the stack. You will have access to thousands of GPUs in this team.

YOUR MISSION

To push the frontier of reasoning and coding capabilities of foundational models, via Reinforcement Learning.

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

Skills & Experience

  • 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)
PROCESS

  • Intro call with one of our Founding Engineers
  • Technical Interview(s) with one of our Founding Engineers
  • Team fit call with the People team
  • Final interview with one of our Founding Engineers

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


Similar Jobs

Explore other opportunities that match your interests

Software Engineer - Avigilon Alta Video UI

Programming
•
4h ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

Silvus Technologies

United Kingdom
Visa Sponsorship Relocation Remote
Job Type Contract
Experience Level Mid-Senior level

Leap29

United Kingdom

Senior Unity Developer (Poker)

Programming
•
7h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Mid-Senior level

Jobgether

United Kingdom

Subscribe our newsletter

New Things Will Always Update Regularly