Staff Performance Optimization Engineer, AI Infrastructure
Description For Staff Performance Optimization Engineer, AI Infrastructure
Tesla is seeking a Staff Performance Optimization Engineer to join their AI Infrastructure team, playing a crucial role in enhancing the efficiency and scalability of Tesla's AI systems. This position offers an exciting opportunity to work at the intersection of high-performance computing and artificial intelligence, directly impacting Tesla's autonomous driving and robotics initiatives.
The role involves collaborating with cross-functional teams to optimize machine learning model training and deployment, focusing on sophisticated performance analysis and optimization of GPU-CPU interactions. You'll be responsible for developing instrumentation tools, troubleshooting complex systems, and implementing data-driven improvements to accelerate AI development pipelines.
As an ideal candidate, you'll bring deep expertise in GPU programming, low-level software development, and performance analysis. Your work will directly contribute to advancing Tesla's ambitious technological goals in the automotive and AI sectors. The position offers competitive compensation ranging from $140,000 to $348,000 annually, plus comprehensive benefits including medical, dental, vision coverage, and various financial benefits.
Working at Tesla's Palo Alto location, you'll be part of a dynamic, fast-paced environment where innovation and collaboration are paramount. This role presents an exceptional opportunity for someone passionate about high-performance computing and AI optimization to make a significant impact on the future of autonomous technology and sustainable transportation.
Responsibilities For Staff Performance Optimization Engineer, AI Infrastructure
- Work with various teams to accelerate time-to-market for new ML models
- Design, implement, and deploy low-overhead instrumentation methods for troubleshooting performance issues
- Analyze collected telemetry, identifying bottlenecks and designing practical solutions
- Develop data-driven performance improvements to existing software pipelines
- Work with teams to validate performance improvements and incorporate them into production training runs
- Help make Tesla's ambitious AI-related products and services a reality
Requirements For Staff Performance Optimization Engineer, AI Infrastructure
- Deep understanding of GPU-based training and inferencing workloads
- Knowledge of CUDA/ZLUDA, OpenCL, PyTorch, Tensorflow, GPUDirect and RDMA-enabling services
- Experience developing and tuning low-level software using C, x86 assembly, and Rust
- Experience with performance analysis techniques (profiling, tracing, simulative analysis)
- Excellent spoken and written communication skills
- Passion for high-performance computing
Benefits For Staff Performance Optimization Engineer, AI Infrastructure
- Medical plans with $0 payroll deduction
- Family-building, fertility, adoption and surrogacy benefits
- Dental and vision plans
- Health Savings Account (HSA) with company contribution
- Healthcare and Dependent Care FSA
- LGBTQ+ care concierge services
- 401(k) with employer match
- Employee Stock Purchase Plans
- Life, AD&D, short-term and long-term disability insurance
- Employee Assistance Program
- Sick and Vacation time
- Back-up childcare and parenting support
- Commuter benefits
- Employee discounts and perks program
Jobs Related To Tesla Staff Performance Optimization Engineer, AI Infrastructure
