Tesla · 2 months ago
Software Engineer, AI Performance Modeling & Co Design
Palo Alto, CA
Full-time
Onsite
Mid Level
$132K/yr - $390K/yrTesla is dedicated to developing and optimizing AI systems that can scale efficiently to thousands of compute nodes. The role involves performance modeling, optimizing AI architectures, and collaborating with hardware teams to ensure seamless integration of AI models with next-generation hardware.
AutomotiveElectric VehicleEnergyManufacturingRenewable EnergyTransportation
Responsibilities
Develop and apply performance modeling techniques to analyze and optimize AI system performance, including dataflow, mapping, and sharding of AI models
Design and implement collective and multi-device parallel computing algorithms and architectures to accelerate AI workloads
Develop and maintain network models and simulations to analyze and optimize AI system performance in distributed environments
Collaborate with cross-functional teams to integrate performance modeling and optimization techniques into AI system design and development workflows
Develop and maintain software tools and frameworks to support performance modeling, parallel computing, and networking development
Work closely with the hardware co-design team to model and optimize future generation AI-ASIC architectures, ensuring seamless integration of AI models and algorithms with next-generation hardware
Co-design and optimize AI systems for future generations, considering emerging trends and technologies, such as 3D stacked processors, photonic interconnects, and novel memory architectures
Develop and apply co-design methodologies to optimize AI system performance, power, and area, while ensuring scalability, reliability, and manufacturability
Qualification
Required
Bachelor's Degree in Computer Science, Computer Engineering, Electrical Engineering, proof of exceptional skills in related fields with practical software engineering experience, or equivalent experience
Experience in performance modeling, parallel computing, and networking, with a focus on AI systems
Proficiency in Python and C++
Experience with AI networks, such as CNNs, transformers, and diffusion model architectures, and their performance characteristics
Experience with parallel computing frameworks and libraries, such as MPI, NCCL, or Gloo
Understanding of GPU and TPU accelerators architecture, memory hierarchy, and ability to model performance and identify performance bottlenecks
Strong understanding of computer networks, including network protocols, topology, and performance modeling
Experience with SW/HW co-design and optimization
Strong analytical and problem-solving skills, with the ability to analyze complex systems and optimize their performance
Benefits
Medical plans > plan options with $0 payroll deduction
Family-building, fertility, adoption and surrogacy benefits
Dental (including orthodontic coverage) and vision plans, both have options with a $0 paycheck contribution
Company Paid (Health Savings Accounts) HSA Contribution when enrolled in the High-Deductible medical plan with HSA
Healthcare and Dependent Care Flexible Spending Accounts (FSA)
401(k) with employer match, Employee Stock Purchase Plans, and other financial benefits
Company paid Basic Life, AD&D
Short-term and long-term disability insurance (90 day waiting period)
Employee Assistance Program
Sick and Vacation time (Flex time for salary positions, Accrued hours for Hourly positions), and Paid Holidays
Back-up childcare and parenting support resources
Voluntary benefits to include: critical illness, hospital indemnity, accident insurance, theft & legal services, and pet insurance
Weight Loss and Tobacco Cessation Programs
Tesla Babies program
Commuter benefits
Employee discounts and perks program
Company
Tesla
Tesla is an electric vehicle and clean energy company that provides electric cars, solar, and renewable energy solutions.
Founded in 2003
10001+ employees
H1B Sponsorship
Tesla has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2796)
2024 (2113)
2023 (1304)
2022 (1227)
2021 (1102)
2020 (723)
Funding
Current Stage
Public CompanyTotal Funding
$20.12BKey Investors
Societe GeneraleEuropean UnionPennDOT
2024-10-05Post Ipo Debt· $783M
2023-09-13Grant· $159.6M
2023-08-15Grant· $0.23M
Leadership Team
Vaibhav Taneja
Chief Financial Officer
Tom Zhu
SVP of Automotive
Recent News
2026-01-09
2026-01-09
2026-01-09
Company data provided by crunchbase