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