Member of Technical Staff - ML Research

Architect Labs is a frontier AI lab specializing in chip design, focusing on co-designing custom ASICs alongside evolving ML workloads. The role involves training AI models for chip design, verification, and exploration, with responsibilities including implementing reinforcement learning environments and collaborating with research teams to translate techniques into production-ready implementations.

Responsibilities:

• Responsible for co-designing and implementing the Reinforcement Learning environments and algorithms, Reward Models trainings and reward signal experiments
• You will work at the intersection of cutting-edge research and production engineering for chip designs, implementing, scaling, and improving post-training techniques to enhance model capabilities and usability
• Design, build, and run robust, efficient pipelines for model fine-tuning and evaluation, ensuring that theoretical performance translates into production-ready implementations
• This is a hands-on, 0→1 role where you'll own the end-to-end RL workflow—from reward modeling and environment design to test-time optimization and scaling
• Collaborate with research teams to translate emerging techniques into production-ready implementations and debug complex issues in training pipelines and model behavior