Member of Technical Staff - Research Intern

About the Company

Architect is an AI research and product lab for chip design. Born out of Stanford, our team blends researchers and engineers from Anthropic, DeepMind, Meta, Apple, Intel, and other frontier labs to reimagine chip design using AI.

Responsibilities

• Co-design and implement Reinforcement Learning experiments (GRPO/PPO/DPO), training data mixes, and reward signal explorations
• Contribute to research on post-training techniques and run ablation studies to improve model reasoning and alignment
• Implement and test new algorithms for model fine-tuning and evaluation
• Analyze experimental results and debug model behavior to establish training best practices

Requirements

• Currently pursuing a PhD or Master’s degree in Computer Science, Machine Learning, Mathematics, or a related field
• Strong academic understanding or project experience with Reinforcement Learning (PPO, DPO, GRPO)
• Strong proficiency in Python and PyTorch
• Experience with training or fine-tuning Large Language Models (LLMs) or the modern NLP stack (Transformers, HuggingFace)
• Ability to read and implement concepts from recent research papers

Preferred Qualifications

• Previous internship experience at frontier AI labs or research organizations
• Publications in top ML venues (NeurIPS, ICLR, ICML) or EDA venues (DAC, ICCAD)
• Familiarity with hardware design concepts such as Verilog, RTL, or EDA tools

Benefits

• Competitive internship stipend
• Mentorship from a team of researchers and engineers from Anthropic, DeepMind, Meta, and Stanford
• Opportunity to work on 0→1 problems in AI-driven chip design