Research Engineer

About the role

As a Research Engineer, you’ll work on training, evaluating, and serving large AI models and new inference-time compute techniques, build internet-scale datasets, and help prototype new research and product ideas.

What you'll work on

• Optimize inference throughput for novel model architectures
• Contribute to frameworks used for our research and production workloads
• Train trillion-parameter models on large GPU clusters
• Curate post-training datasets to improve targeted capabilities
• Build out internet-scale data pipelines and crawlers
• Design, prototype, and optimize new model architectures
• Contribute to research across long-context, inference-time compute, RL, and more

What we’re looking for

• Strong general software engineering skills
• Thorough knowledge of the deep learning literature
• Experience with pre- and post-training of LLMs
• Ability to come up with and evaluate research ideas
• Experience working with large distributed systems
• Ability to handle large ETL workloads

Compensation, benefits, and perks (US)

• Annual salary ranges between $275K - $550K based on experience