Machine Learning – Research Engineer – Horizons

Machine Learning – Research Engineer – Horizons

Requirements

• Are proficient in Python and async/concurrent programming with frameworks like Trio
• Have experience with machine learning frameworks (PyTorch, TensorFlow, JAX)
• Have industry experience in machine learning research
• Can balance research exploration with engineering implementation
• Enjoy pair programming (we love to pair!)
• Care about code quality, testing, and performance
• Have strong systems design and communication skills
• Are passionate about the potential impact of AI and are committed to developing safe and beneficial systems

Responsibilities

• Architect and optimize core reinforcement learning infrastructure, from clean training abstractions to distributed experiment management across GPU clusters. Help scale our systems to handle increasingly complex research workflows.
• Design, implement, and test novel training environments, evaluations, and methodologies for reinforcement learning agents which push the state of the art for the next generation of models.
• Drive performance improvements across our stack through profiling, optimization, and benchmarking. Implement efficient caching solutions and debug distributed systems to accelerate both training and evaluation workflows.
• Collaborate across research and engineering teams to develop automated testing frameworks, design clean APIs, and build scalable infrastructure that accelerates AI research.

Preferred Qualifications

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