Founding ML Engineer, RL for Custom ASICs

Architect Labs; Palo Alto (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Competitive salary

Equity stake

Job summary

Architect is a frontier AI lab focused on chip design, building AI models and tools for on-demand ASICs. As a Founding Member of the Technical Staff, you will train AI models for chip design, verification, and exploration, with production-scale deployment at leading foundries like TSMC.

You will co-design RL environments, implement post-training techniques, and build end-to-end ML pipelines that translate research into reliable, production-ready systems.

Qualifications

• PhD in CS/EECS or related field, with ML/AI specialization.
• Deep RL expertise and post-training deployment experience.
• End-to-end ML pipeline development experience including RL and fine-tuning LLMs.
• Experience with large-scale distributed ML systems and training frameworks.
• Strong debugging and engineering rigor; fast-moving builder.

Responsibilities

• Co-design and implement RL environments and reward models.
• Work at intersection of research and production for chip designs.
• Design, build, and run robust pipelines for model fine-tuning and evaluation.
• Own end-to-end RL workflow from rewards to test-time optimization.
• Collaborate with research teams to translate techniques into production.

Skills

Reinforcement Learning

LLM fine-tuning

Distributed systems

PyTorch

CUDA

Code models

Education

PhD in Computer Science or related field

BS/MS with strong research engineering background

Tools

PyTorch

CUDA

QLoRA

ZeRO

Job description

Architect is a frontier AI lab focused on chip design, building AI models and tools for on-demand ASICs. As a Founding Member of the Technical Staff, you will train AI models for chip design, verification, and exploration, with production-scale deployment at leading foundries like TSMC.

You will co-design RL environments, implement post-training techniques, and build end-to-end ML pipelines that translate research into reliable, production-ready systems.