Senior ML Engineer (AI Research)

📋 Description

• Conduct applied machine learning research across areas such as guided search, reinforcement
• Design and execute experiments to identify efficient approaches for training large language models
• Explore methods for guided generation and search within model trajectory spaces
• Investigate approaches for collecting and mining relevant training data at web scale
• Develop efficient methods for incorporating large-scale data into model post-training workflows
• Experiment with different reinforcement learning configurations in domains where rewards can be

🎯 Requirements

• Significant professional experience in machine learning engineering, applied AI research, or a
• Profound understanding of the theoretical foundations of machine learning and reinforcement learning
• Deep expertise in modern deep learning for language processing and generation
• Substantial experience training large-scale models across multiple computational nodes
• Strong software engineering capabilities, particularly with Python
• Deep hands-on experience with modern deep learning frameworks, particularly JAX

🎁 Benefits

• Competitive compensation
• Career growth and continuous learning opportunities
• Flexibility and significant ownership in your work
• Collaborative and innovative research and engineering culture
• Opportunity to work on impactful AI research and development projects
• Exposure to advanced machine learning, reinforcement learning, LLMs, and agentic AI systems