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 Research Engineer USD 177K-251KBenchmarks | Data Pipelines | Data Versioning | Evaluation | Fine TuningCross-functional collaboration | End-to-end ownership | High autonomyMid-level Full TimeBellevue, WA | Menlo Park, CA …3h ago
-
Data Engineer USD 185K-196KApache Spark | Artificial Intelligence | CSS | Data Governance | Data ModelingMid-level Full TimeMenlo Park, CA3h ago
-
Entry-level Full TimeMenlo Park, CA3h ago
-
Privacy Engineer USD 194K-217KApache Airflow | Apache Spark | Automated testing | C plus plus | Continuous DeploymentEntry-level Full TimeMenlo Park, CA3h ago
-
Software Engineer III, AI/ML GenAI, Google Cloud Compute USD 147K-211KAudio generation | C++ | Computer Vision | Data Processing | Data StorageSenior-level Full TimeKirkland, WA, USA3h ago
-
Senior Software Engineer, AI/ML, Google Cloud AI USD 174K-252KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Software Engineer, TPU Inference, AI/ML USD 147K-211KCloud Computing | Compilers | GPU | GPU Programming | InferenceMid-level Full TimeKirkland, WA, USA3h ago
-
Backup and Recovery | BigQuery | C++ | Consistency models | Copy-on-writeSenior-level Full TimeKirkland, WA, USA3h ago
-
Software Engineer, Machine Health USD 147K-211KAnalysis and Design | C++ | Data Processing | Data analytics | Distributed SystemsMid-level Full TimeSunnyvale, CA, USA3h ago
-
Networking AI Technical Lead USD 207K-300KAI | Algorithms | C++ | Compute Technologies | Data StructuresSenior-level Full TimeSunnyvale, CA, USA; Cambridge, MA, USA3h ago
-
Staff Software Engineer, BigQuery Managed Storage USD 207K-300KApache Iceberg | Data Lakes | Data Structures | Data Structures and Algorithms | Distributed ComputingSenior-level Full TimeKirkland, WA, USA; Sunnyvale, CA, USA3h ago
-
Software Engineer III, AI/ML, Health and Home USD 147K-211KData Analysis | Data Processing | Data Storage | Debugging | Distributed ComputingSenior-level Full TimeMountain View, CA, USA3h ago
-
Customer Engineer II, Applied AI, Google Cloud USD 149K-216KApplication Architecture | Artificial Intelligence | C++ | Cloud Computing | Cloud platformSenior-level Full TimeNew York, NY, USA; Chicago, IL, …3h ago
-
Software Engineer III, AI/ML GenAI, Google Cloud AI USD 147K-211KC++ | Data Processing | Debugging | Generative AI | Language ModelsSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Senior Software Engineer, Full Stack, Gen AI Formats USD 174K-252KArtificial Intelligence | C++ | Data Storage | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeMountain View, CA, USA3h ago
-
Customer Engineer, Data Analytics, Google Cloud USD 153K-222KBatch Processing | Big Data | Cloud Native | Cloud Native Architecture | Cloud platformSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Senior Software Engineer, Managed Kafka USD 174K-252KApache Kafka | Cloud platform | Distributed Systems | Go | Google CloudSenior-level Full TimeNew York, NY, USA; Raleigh, NC, …3h ago
-
Senior Software Engineer, AI/ML, Retail Ads, Fullstack USD 174K-252KAlgorithms | C++ | Data Processing | Data Structures | DebuggingSenior-level Full TimeMountain View, CA, USA3h ago
-
Algorithms | Data Structures | Deep learning | Embeddings | Information RetrievalSenior-level Full TimeMountain View, CA, USA3h ago
-
Senior Software Engineer, AI Infra, Ads Safety USD 174K-252KBenchmarking | C++ | Data Processing | Data Storage | Data platformSenior-level Full TimeLos Angeles, CA, USA3h ago
-
AWS | Azure | CI/CD | Data Engineering | DockerCross-functional collaboration | Documentation support | Flexible teamwork environment | MentorshipEntry-level InternshipIrvine, CA, US6h ago
-
Senior Software Engineer, Data Platform & AI Enablement USD 158K-260KAI infrastructure | Checkpointing | Distributed Systems | Exactly once | Exactly-once semanticsSenior-level Full TimeUS - San Francisco8h ago
-
Staff Backend Software Engineer- (AI Platform) USD 192K-260KAPI Development | Algorithms | Autoscaling | Data Structures | Distributed SystemsSenior-level Full TimeSan Francisco, California10h ago
-
Staff Backend Software Engineer- (AI Platform) USD 166K-225KAPI Design | Algorithms | Autoscaling | Caching | Data StructuresSenior-level Full TimeSan Francisco, California10h ago
-
Staff Backend Software Engineer- (AI Platform) USD 192K-260KAPI Design | APIs | Algorithms | Autoscaling | CachingSenior-level Full TimeSan Francisco, California10h ago