Research Engineer - Agency and Reasoning

The role

You will be a core contributor to the Agency and Reasoning Team. You will perform novel research in reinforcement learning, post-training, and human preference learning and apply ideas to next generation language models.

What you will do

• Perform novel research in reinforcement learning, post-training, and human preference learning
• Apply ideas at scale to next generation language models
• Work through a research project from conception to execution to write-up
• Take an idea from conception to experimentation extremely quickly
• Work and cooperate with others in a high-paced research setting
• Work effectively on both research and engineering implementation at scale

Required qualifications

• Strong research taste and intuition
• The ability to work through a research project from conception to execution to write-up
• Strong implementation and prototyping skillset
• A researcher who 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
• Postgraduate degree in a scientific subject (Computer Science, EE/EECS, Mathematics, Physics)
• Previously published machine learning research in well-respected venues
• Highly proficient with PyTorch and Python
• Excellent communication and collaboration skills

Nice to have

• Experience and aptitude with reinforcement learning, either in the context of language model reasoning or more classical RL tasks
• Experience with language-model-supervised fine-tuning and preference-learning methods, such as DPO and simPO
• Experience with context-length extension methods
• A good intuitive ability to understand model behaviors and correct them through iterative fine-tuning
• Interest in grappling in detail with data and spending significant time involved in data engineering and synthetic data generation
• We are excited and able to rapidly learn new fields and implement new ideas

Benefits

• 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

About the company

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