Zyphra
Research Engineer - Language Model Pre-Training
San Francisco
Sponsorship not specifiedDetected 127 days ago
Node.jsMachine LearningPyTorchNLPA/B TestingResearchExperimental DesignCommunicationCollaboration
About the role
- As a Research Engineer - Language Model Pre-Training, you'll shape our language model roadmap through end-to-end pretraining development.
- You will work extremely closely with our pretraining team, who will integrate your insights into our next-generation models.
- Large-scale training runs and model parallelization
Responsibilities
- Can rapidly learn new fields and are excited to implement new ideas
- Relocation and immigration support on a case-by-case basis
Requirements
- Our research methodology is to make grounded, methodical steps toward ambitious goals. Both deep research and engineering excellence are equally valued
Skills
- 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.
- Strong 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)
- 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 bar to impact as low as possible
- We all enjoy what we do and love discussing AI
Compensation
- Competitive compensation and 401(k) plan
Benefits
- Comprehensive medical, dental, vision, and FSA plans
- Unlimited PTO and company holidays
Company info
- ZYPHRA IS AN ARTIFICIAL INTELLIGENCE COMPANY BASED IN SAN FRANCISCO, CALIFORNIA.
This listing is sourced directly from Zyphra's careers page and normalized into a canonical job model.