Research Engineer - Language Model Pre-Training

Highlights

• Shape Zyphra's language model roadmap via end-to-end pretraining development.
• Drive large-scale training runs and advance model parallelization techniques.
• Lead dataset collection, processing, and evaluation for pretraining.

What they offer

• Comprehensive medical, dental, vision, and FSA plans
• Competitive compensation and 401(k) plan
• Relocation and immigration support on a case-by-case basis
• In-office snacks and meals provided
• Unlimited PTO and company holidays

Ideal candidate

A fast-learning builder who blends research rigor with engineering excellence and thrives collaborating at scale. You love bold ideas, grounded experimentation, and turning insights into next-generation models.

Requirements

• Strong engineering aptitude for rapidly implementing reliable and robust systems
• Ability to rapidly learn new fields and implement new ideas
• Excellent communication and collaboration skills across research and engineering at scale
• Deep expertise and intuition for solving machine learning problems and training models
• Experience training on large-scale multi-node GPU clusters
• Deep understanding of model training pipelines including model/data parallelism and distributed optimizers
• Strong grasp of experimental methodology for rigorous ablations and hypothesis testing
• Understanding of large-scale, highly parallel data processing pipelines
• High proficiency with PyTorch and Python
• Ability to dive into large pre-existing codebases and get up to speed quickly
• Postgraduate degree in a scientific subject such as Computer Science, EE/EECS, Math or Physics