Research Engineer / Scientist, Pre-training

Conduct research and implement solutions in areas such as model architecture, algorithms, data processing, and optimizer development.

Independently lead small research projects while collaborating with team members on larger initiatives.

Design, run, and analyze scientific experiments to advance our understanding of large language models.

Optimize and scale our training infrastructure to improve efficiency and reliability.

Advanced degree (MS or PhD) in Computer Science, Machine Learning, or a related field.

Strong software engineering skills with a proven track record of building complex systems.

Expertise in Python and experience with deep learning frameworks (PyTorch preferred).

Familiarity with large-scale machine learning, particularly in the context of language models.