Software Engineer, Performance Modeling, Dojo

Job Description

As a member of the Dojo Machine Learning team, you'll develop and optimize simulations for a massively parallel machine for AI training. Responsibilities include validating simulations of architecture, collaborating with engineers, maintaining software frameworks, and ensuring seamless integration with other system components. Stay updated with AI workloads and computer architecture to deliver high-performance simulations.

Requirements

• Degree in Engineering, Computer Science, or equivalent experience
• 3+ years in simulation development and AI workloads
• Experience with analytical and cycle-based simulation techniques
• Strong programming skills in C++ and Python
• Experience with deep learning frameworks like PyTorch and JAX
• Understanding of CPU/GPU microarchitecture.

Responsibilities

• Develop and validate simulations of AI training architecture
• Collaborate with architects
• Maintain software frameworks and tools
• Ensure integration of simulations
• Participate in code reviews and debugging
• Stay updated with developments in AI workloads and computer architecture.

Benefits

• Benefits include Aetna PPO and HSA plans with $0 payroll deduction, family-building benefits, dental and vision plans with $0 contribution, flexible spending accounts, 401(k) with employer match, company-paid life insurance, sick and vacation time, and employee discounts.