Research Engineer - Model Architectures

As a Research Engineer - Model Architectures, you will be a core contributor to Zyphra’s AI Architecture Research Team. This will involve designing and rigorously testing novel model architectures and training methodologies, with a focus on improving core modeling capabilities (e.g., loss per flop or loss per parameter) and addressing fundamental bottlenecks in contemporary models. You will also work extremely closely with our pre-training team, who will integrate your insights into our next-generation models.

• Strong research taste and intuition

• The ability to work through a research project from conception to execution to write-up

• Strong implementation and prototyping ability can take an idea from conception to experimentation extremely quickly

• The ability to work well and cooperate with others in a high-paced research setting

• Curiosity, interest, and joy in understanding intelligence.

• Previous experience with long-term memory, RAG/retrieval systems, dynamic/adaptive computation, and alternative approaches to credit assignment

• Experience with reinforcement learning, control theory, and signal processing

• Generally, a joy in inventing and seriously assessing ‘crazy’ ideas, and the ability to have a unique perspective on things

• Understanding of modern training pipelines and the hardware requirements to design efficient architectures for GPU hardware

• Strong grasp of proper experimental methodology for running rigorous ablations and other hypothesis testing

• High proficiency with PyTorch and Python.

• Strong ability to jump into large pre-existing codebases and rapidly get up to speed and become productive

• Previously published machine learning research in well-respected venues

• Postgraduate degree in a scientific subject (Computer Science, EE/EECS, Math, Physics)