What to Expect
As a member of the Dojo Machine Learning team, you will be responsible for developing and optimizing simulations of the architecture of a massively parallel machine for AI training. The ideal candidate will have a strong background in computer architecture, analytical and cycle-based simulation, and AI workloads, with a passion for delivering high-performance simulations that accurately model complex systems. 

What You’ll Do
Develop and validate simulations of the architecture of a massively parallel machine for AI training, including system architecture, core architecture, memory hierarchy, and interconnects 
Collaborate with architects and engineers to understand the requirements of the simulation and ensure that it accurately models the behavior of the system 
Develop and maintain software frameworks and tools to support simulation development, testing, and deployment 
Work closely with the team to ensure seamless integration of simulations with other components of the system 
Participate in code reviews, testing, and debugging to ensure high-quality software 
Stay up-to-date with the latest developments in AI workloads, computer architecture, and simulation techniques 
What You’ll Bring
Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability  
3+ years of experience in simulation development, computer architecture, and AI workloads 
Experience with analytical and cycle-based simulation techniques, including modeling of complex systems and validation of simulation results 
Strong programming skills in languages such as C++ and Python 
Experience with deep learning frameworks such as PyTorch and JAX  
Strong understanding of CPU and/or GPU microarchitecture, including pipelining, caching, and memory hierarchy 
Excellent problem-solving skills, with the ability to analyze complex problems and develop creative solutions 
Strong communication and collaboration skills, with the ability to work effectively with Architects, Engineers, and Researchers 
Able to work from Palo Alto office

PALO ALTO, California

Full time