Research Engineer, Machine Learning (Reinforcement Learning)

Requirements:

• Proficient in Python and async/concurrent programming with frameworks like Trio
• Experience with machine learning frameworks (PyTorch, TensorFlow, JAX)
• Industry experience in machine learning research
• Balance research exploration with engineering implementation
• Enjoy pair programming
• Care about code quality, testing, and performance
• Strong systems design and communication skills
• Passionate about the potential impact of AI and committed to developing safe and beneficial systems

Nice to Haves:

• 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++

What You'll Be Doing:

• Collaborating with researchers and engineers to advance large language model capabilities and safety
• Working on fundamental research in reinforcement learning
• Designing, implementing, and testing novel training environments and methodologies
• Driving performance improvements through profiling, optimization, and benchmarking
• Developing automated testing frameworks and scalable infrastructure

Perks and Benefits:

• Competitive compensation and benefits
• Optional equity donation matching
• Generous vacation and parental leave
• Flexible working hours
• Lovely office space for collaboration