Founding Member of Technical Staff - Post Training
Architect Labs· Engineering
Palo Alto, CAFull-timePosted 7mo ago
About the role
A role at the forefront of AI model post-training for chip design tasks, involving the development and scalability of reinforcement learning environments and algorithms for hardware design automation.
Responsibilities
- Responsible for co-designing and implementing the Reinforcement Learning environments and algorithms, Reward Models trainings and reward signal experiments.
- Work at the intersection of research and production engineering for chip designs, implementing, scaling, and improving post-training techniques.
- Design, build, and run robust, efficient pipelines for model fine-tuning and evaluation, ensuring performance translates into production-ready implementations.
- 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.
Qualifications
- PhD in Computer Science, EECS, Mathematics, or a related field, preferably with specialization in Machine Learning, Deep Learning, or AI.
- Deep expertise in reinforcement learning and post-training, with a proven track record of deploying models from research to real-world.
- Strong industry or research background building end-to-end ML pipelines, especially RL and fine-tuning LLMs and code models.
- Strong software engineering skills with experience in large-scale distributed systems, high-performance computing, and frameworks like PyTorch, CUDA, QLoRA, ZeRO.
Skills
Required:Reinforcement LearningMachine Learning PipelinesSystems Engineering
Benefits
- Competitive salary and meaningful equity stake.
- Fast-paced startup environment with autonomy and visible impact.
- Cutting-edge AI-driven chip design challenges.
Last updated Mar 31, 2026. Interested? Apply on the company site.
About Architect Labs
Architect Labs
Architect is a frontier AI lab for chip design.
More open roles at Architect Labs
- Member of Technical Staff - Physical Design1mo agoPalo Alto · ASIC
- Member of Technical Staff - Formal Verification2mo agoPalo Alto · ASIC
- Microarchitect / RTL Design - Memory Subsystem2mo agoPalo Alto, Bangalore, India · Architecture
- Microarchitect / RTL Design - Interconnect & Fabric2mo agoPalo Alto · Architecture
- Member of Technical Staff - Microarchitectural Validation3mo agoBangalore, India · ASIC