AI Compute Architect, AI Hardware

What To Expect

The Tesla AI Hardware team is dedicated to advancing artificial intelligence through innovative hardware solutions. Key projects include Dojo, a custom supercomputer designed to train large-scale neural networks using video data from Tesla vehicles. This team's work powers the neural networks behind Full Self-Driving (FSD) and the company's humanoid robot, Optimus, pushing the boundaries of computational efficiency and performance.

What You'll Do

• Design and architect comprehensive end-to-end compute infrastructure for AI/ML pipelines, including hardware specifications, system topologies, and custom accelerators
• Collaborate with system architects, micro architects, IP vendors, and program management to guide SoC development from concept to production
• Optimize the interplay between compute, storage, and interconnect to maximize throughput, minimize latency, and reduce energy consumption
• Partner with ML model designers, compiler engineers, and software developers to build intuitive tools, frameworks, and abstractions for AI/ML workload deployment and scaling
• Evaluate emerging technologies and prototype innovative architectures to address evolving AI compute demands
• Maintain expertise in the latest advancements in AI workloads, domain-specific languages, computer architecture principles, and advanced simulation methodologies

What You'll Bring

• Degree in Electrical Engineering, Computer Science, Computer Engineering, or a related field; or equivalent practical experience
• Deep knowledge of CPU, GPU, and ML accelerator microarchitectures
• Strong understanding of Large Language Models (LLMs) and techniques for their training, inference, quantization, and optimization
• Exceptional problem-solving abilities and a track record of devising innovative solutions
• Hands-on experience with deep learning frameworks and proficiency in performance analysis using simulation frameworks

Benefits

• Competitive pay and benefits package including medical, dental, vision, and flexible spending accounts
• Healthcare and dependent care flexible spending accounts
• Short-term and long-term disability insurance
• Employee assistance program
• Paid holidays, sick leave, and vacation time
• Voluntary benefits including critical illness, hospital indemnity, accident insurance, theft & legal services, and pet insurance
• Weight loss and tobacco cessation programs
• Employee discounts and perks program