ZyphraZyphra is an artificial intelligence company based in San Francisco, California.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.You'll Work Across:Large-scale training runs and model parallelizationPerformance optimization of our pretraining stackDataset collection, processing, and evaluationArchitecture and methodology research, including optimizer ablationsWhat We're Looking For / Requirements:Strong engineering aptitude for rapidly implementing reliable and robust systemsCan rapidly learn new fields and are excited to implement new ideasExcellent communication and collaboration skills, and can work effectively on both research and engineering implementation at scaleQualifications / Additional Skills:Deep expertise and intuition for solving machine learning problems and training modelsExperience with training on large-scale (multi-node) GPU clustersDeep 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 testingUnderstanding of large-scale, highly parallel data processing pipelinesHigh proficiency with PyTorch and Python.Strong ability to dive into large pre-existing codebases and rapidly get up to speedPublished machine learning research in well-respected venues is a plusPostgraduate degree in a scientific subject (Computer Science, EE/EECS, Math, Physics)Why Work at Zyphra:Our research methodology is to make grounded, methodical steps toward ambitious goals. Both deep research and engineering excellence are equally valuedWe strongly value new and crazy ideas and are very willing to bet big on new ideasWe move as quickly as we can; we aim to minimize the bar to impact as low as possibleWe all enjoy what we do and love discussing AIBenefits and Perks:Comprehensive medical, dental, vision, and FSA plansCompetitive compensation and 401(k) planRelocation and immigration support on a case-by-case basisIn-office snacks and meals providedUnlimited PTO and company holidaysIn-person team in San Francisco with a collaborative, high-energy environment