Opening. This role is a key part of our mission to create beneficial AI systems. We're looking for a Research Engineer to join our Horizons team, which leads our reinforcement learning research and development.
What you'll do
As a Research Engineer on the Horizons team, you will collaborate with a diverse group of researchers and engineers to advance the capabilities and safety of large language models. This role blends research and engineering responsibilities, requiring you to both implement novel approaches and contribute to the research direction.
- Develop and implement novel reinforcement learning algorithms and techniques to improve the performance and safety of our AI systems.
- Collaborate with researchers and engineers to design and implement experiments to test and evaluate the effectiveness of our AI systems.
- Work with the applied production training team to bring research innovations into deployed models.
What you need
- Proficiency in Python and async/concurrent programming with frameworks like Trio.
- Experience with machine learning frameworks (PyTorch, TensorFlow, JAX).
- Familiarity with LLM architectures and training methodologies.
- Experience with reinforcement learning techniques and environments.
- Experience with virtualization and sandboxed code execution environments.
- Experience with Kubernetes.
- Experience with distributed systems or high-performance computing.
- Experience with Rust and/or C++.
Why this matters
Our research has the potential to significantly impact the development of AI systems and their applications in various fields. By advancing the capabilities and safety of large language models, we can create more beneficial and reliable AI systems that can be used to improve people's lives.