Research Engineer - Model Architectures

Zyphra; San Francisco (CA); On-site; USD 100,000 - 150,000; Full time

Benefits offered by this job

Comprehensive medical, dental, and vision plans

Competitive compensation and 401(k) plan

Relocation and immigration support

In-office snacks and meals

Unlimited PTO and company holidays

Collaborative work environment in San Francisco

Job summary

A leading AI research firm in San Francisco is seeking a Research Engineer specializing in Model Architectures. You will design and rigorously test innovative model architectures, improving core modeling capabilities and collaborating closely with pre-training teams. Candidates should exhibit strong research intuition, proficiency in PyTorch and Python, and have a postgraduate degree in a relevant scientific discipline. The firm values both deep research and engineering excellence, making it an exciting place for innovative thinkers.

Qualifications

• Experience with long-term memory and retrieval systems.
• Familiarity with reinforcement learning and control theory.
• Joy in inventing and assessing unconventional ideas.
• Understanding modern training pipelines for GPU designs.
• Ability to conduct rigorous hypothesis testing.
• Experience with large pre-existing codebases.

Responsibilities

• Design novel model architectures for AI.
• Rigorous testing of training methodologies.
• Collaborate with pre-training teams for model integration.

Skills

Strong research taste and intuition

Ability to work through a research project

Strong implementation and prototyping ability

Cooperation in high-paced settings

Curiosity about intelligence

Education

Postgraduate degree in a scientific subject

Tools

PyTorch

Python

Job description

Zyphra is an artificial intelligence company based in San Francisco, California.

The Role:

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.

What We're Looking For / Requirements:

• 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.

Qualifications / Additional Skills:

• 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)

Why Work at Zyphra:

• Our research methodology is to make 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 bar 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) plan

• Relocation and immigration support on a case-by-case basis

• In-office snacks and meals provided

• Unlimited PTO and company holidays

• In-person team in San Francisco with a collaborative, high-energy environment