Research Engineer, Machine Learning (Reinforcement Learning)

📋 Description

• Collaborate with researchers and engineers to advance capabilities and safety of RL for LLMs.
• Blend research and engineering to implement novel approaches and guide direction.
• Develop agentic models via tool use for open ended tasks like computer use.
• Improve model reasoning in areas such as mathematics and reasoning tasks.
• Architect and scale RL infrastructure and training methodologies.
• Build scalable tooling and APIs to accelerate AI research and deployment.

🎯 Requirements

• Proficient in Python and async/concurrent programming with Trio.
• Experience with ML frameworks PyTorch, TensorFlow, or JAX.
• Industry experience in machine learning research.
• Balance research exploration with engineering implementation.
• Enjoy pair programming and collaboration.
• Strong systems design and communication skills.

🎁 Benefits

• Competitive compensation and benefits.
• Optional equity donation matching.
• Generous vacation and parental leave.
• Flexible working hours.
• Office space in London.