Staff Software Engineer, TPU Performance
Tasks
- Analyze performance and efficiency metrics
- Apply model co design quantization and sparsity techniques
- Collaborate with product teams to onboard ML models onto TPU hardware
- Design and implement performance solutions
- Identify bottlenecks at fleet wide scale
- Improve compiler and runtime performance
- Maintain ML training and serving benchmarks
- Optimize performance and throughput
- Perform TPU fleet efficiency analysis
- Use benchmarks to identify performance opportunities
Perks/Benefits
- N/A
Skills/Tech-stack
Algorithm Design | Benchmarking | Compiler | Data Analysis | Data Structures | Data Structures and Algorithms | Debugging | Distributed Systems | Hardware-aware algorithm design | JAX | Low-level programming | Machine Learning | Memory hierarchy | OpenXLA | Performance Modeling | Performance optimization | PyTorch | Quantization | Runtime | Software Architecture | Sparsity | Tensor Processing Unit | Tensor processing | Visualization Tools
Education
Roles
Regions
Countries
States
Cities
Related jobs
-
AI Agents | C++ | Data Structures | Data Structures and Algorithms | Generative AISenior-level Full TimeNew York, NY, USA2h ago
-
Accelerated inference | Algorithms | Android | CPU | ChromeSenior-level Full TimeSunnyvale, CA, USA2h ago
-
ML Engineer, Query Understanding, Search Ads USD 174K-252KC++ | Data Processing | Data Storage | Debugging | Deep learningMid-level Full TimeMountain View, CA, USA2h ago
-
Customer Engineer, Data Analytics, Financial Services USD 153K-222KApache Spark | Batch Processing | Big Data | Cloud Computing | Cloud platformSenior-level Full TimeNew York, NY, USA; Atlanta, GA, …2h ago
-
BigQuery | Cloud platform | Data Processing | Deep learning | Distributed SystemsMid-level Full TimeKirkland, WA, USA2h ago
-
Software Engineer, Machine Management Qual Infra USD 147K-211KArtificial Intelligence | C++ | Fleet Management | Machine Learning | Software developmentMid-level Full TimeSunnyvale, CA, USA2h ago
-
Staff Software Engineer, AI/ML, YouTube USD 207K-300KData Processing | Data Structures | Data Structures and Algorithms | Debugging | Distributed SystemsSenior-level Full TimeSan Bruno, CA, USA; Mountain View, …2h ago
-
Software Engineer III, AI/ML, Google Play USD 147K-211KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA2h ago
-
Data Structures | Data Structures and Algorithms | Deep learning | Diffusion Models | Generative AISenior-level Full TimeMountain View, CA, USA2h ago
-
Senior Software Engineer, AI/ML, Payments Platform USD 174K-252KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA2h ago
-
Staff Software Engineer, AI/ML Machine Repairs USD 207K-300KArtificial Intelligence | C++ | Data Processing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Machine Learning Engineer, XLS, Compiler USD 174K-252KAI | C++ | Compiler Construction | Data Processing | Data StructuresMid-level Full TimeChicago, IL, USA2h ago
-
Senior ML/AI Software Engineer, Chrome AI USD 174K-252KAgent Orchestration | Algorithms | Automated testing | C++ | Computer VisionSenior-level Full TimeMountain View, CA, USA2h ago
-
Staff Software Engineer, AI/ML, Agent Assist USD 207K-300KAgentic AI | Cloud Computing | Data Processing | Debugging | Deep learningSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Sr. Data Engineer USD 131K-164KAgentic AI | DBT | Data Governance | Data Quality | Data SecurityConvenient location | Group insurance | Hospitalization insurance | Meal vouchers | Open company cultureSenior-level Full TimeUnited States, FL, US, 5600767h ago
-
Machine Learning Engineer, API Multicloud USD 295K-445KAWS | Data Pipelines | Data Structures | Data structures algorithms | Deep learningSenior-level Full TimeSan Francisco11h ago
-
Agentic AI | Agile | Cloud Computing | Container Orchestration | Distributed SystemsSenior-level Full TimeUnited States12h ago
-
AI Resident USD 120K-120KControl | Habitat | Imitation Learning | Isaac Sim | Isaac-GymCompany-provided meals | Fitness center membership | GPU compute access | Team eventsEntry-level Full TimeMilpitas, CA12h ago
-
Tech Lead, Robotic AI Model USD 150K-180KAction Chunking | Action Tokenization | Behavior Cloning | DPO | DeepSpeedSenior-level Full TimeEl Segundo, California, United States12h ago
-
ML Engineer, Post-Training and Evaluation USD 175K-270KData Pipelines | Data Quality | Data quality analysis | Dataset Preparation | Evaluation methodologyDental insurance | Disability insurance | Health insurance | Life insurance | Paid time offMid-level Full TimeSan Francisco13h ago
-
Lead Gameplay AI Engineer USD 195K-243KC# | C++ | Game AI | Machine Learning | Unreal Engine401k match | Dental insurance | Disability insurance | Flexible PTO | Life insuranceSenior-level Full TimeUnited States (Remote) R13h ago
-
Data analytics | Language Processing | Machine Learning | Natural Language | Natural Language ProcessingCollaborative environment | Mentorship | TrainingSenior-level Full TimeSan Francisco, California, United States - … R13h ago
-
AI Agents | AI Safety | AI Search | AWS | Agentic Workflows401k | Dental insurance | Medical insurance | Paid sick hours | Vision insuranceSenior-level Contract Full TimeRidgefield Park, NJ, United States13h ago
-
Research Engineer, Data Infrastructure USD 175K-225KCloud Native | Cluster Orchestration | Columnar Storage | Data Lakes | Data Lineage401k matching | Coaching | Gym reimbursement | Healthcare | Meal stipendMid-level Full TimePalo Alto14h ago
-
Backend Engineer I - Data Pipelines USD 79K-133KAPI Design | Batch Processing | C# | Distributed Systems | Java401k | Dental insurance | Life insurance | Medical insurance | Paid HolidaysEntry-level Full TimeRedlands, CA14h ago