Zyphra is an artificial intelligence company based in Palo Alto, California.
The Role:
As a Research Engineer specializing in Language Model Pre-training, you'll shape our language model roadmap through end-to-end pre-training development. You will work extremely closely with our pre-training team, who will integrate your insights into our next-generation models.
You will work across:
- Large-scale training runs and model parallelization
- Performance optimization of our pre-training stack
- Dataset collection, processing, and evaluation
- Architecture and methodology research, including optimizer ablations
Requirements:
- Strong engineering aptitude for rapidly implementing reliable and robust systems
- Can rapidly learn new fields and are excited to implement new ideas
- Excellent communication and collaboration skills, and can work effectively on both research and engineering implementation at scale
Ideal Skillset:
- Deep expertise and intuition for solving machine learning problems and training models
- Experience with training on large-scale (multi-node) GPU clusters
- Deep understanding of model training pipelines – including model/data parallelism, distributed optimizers, etc.
- Firm grasp of proper experimental methodology for running rigorous ablations and other hypothesis testing
- Understanding of large-scale, highly parallel data processing pipelines
- High proficiency with PyTorch and Python.
- Strong ability to dive into large pre-existing codebases and rapidly get up to speed
- Published machine learning research in well-respected venues is a plus
- Postgraduate degree in a scientific subject (Computer Science, EE/EECS, Math, Physics)
Why Work at Zyphra:
- Our research methodology involves making grounded, methodical steps toward ambitious goals. Both deep research and engineering excellence are equally valued
- We strongly value new and crazy ideas and are very willing to bet big on new ideas
- We move as quickly as we can; we aim to minimize the barrier to impact as low as possible
- We all enjoy what we do and love discussing AI
Benefits and Perks:
- Comprehensive medical, dental, vision, and FSA plans
- Competitive compensation and 401(k)
- Relocation and immigration support on a case-by-case basis
- On-site meals prepared by a dedicated culinary team; Thursday Happy Hours
- In-person team in Palo Alto, CA, with a collaborative, high-energy environment
We're building something new, and we want someone with strong research taste and intuition. If that is you, Apply Today!
