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
-
Mid-Level Data Engineer USD 90K-98KAPI Development | Azure Data | Azure Data Factory | Azure Data Lake | Azure Data Lake StorageRemote workMid-level Full TimeWork from home, VA, United States R7h ago
-
Senior Data Engineer USD 165K-180KAPIs | Anomaly Detection | Azure | Azure Data | Azure Data FactorySenior-level Full TimeWork from home, VA, United States R7h ago
-
Evergreen - Mathematics for Machine Learning USD 80K-300KAutodiff | JAX | Linear Algebra | Matrix Operations | NumPyAsynchronous hiring process | Flexible collaboration | Part-time hoursMid-level Full TimeBoston, US9h ago
-
Senior Director, AI / Machine Learning Software Engineer USD 136K-300KApache Flink | Apache Spark | CI/CD | Data Lineage | Data PrivacyHealth benefits | Paid leave | Paid volunteer timeSenior-level Full TimeNew York, NY, United States10h ago
-
Computer Vision | Data Analysis | Language Models | Language Processing | Large Language ModelsSenior-level Full TimeSeattle, Washington, United States10h ago
-
Classification Algorithms | Data Analysis | Deep learning | Language Models | Language ProcessingSenior-level Full TimeSan Jose, California, United States10h ago
-
Data Engineering | Machine Learning | Machine Learning Pipelines | Python | Recommendation SystemsSenior-level Full TimeSan Jose, California, United States10h ago
-
Data Pipelines | Full Stack | Full-Stack Development | Machine Learning | PythonSenior-level Full TimeSan Jose, California, United States10h ago
-
C++ | Data Analysis | Data Manipulation | Data Processing | Deep learningSenior-level Full TimeMountain View, CA, USA11h ago
-
Algorithms | Audio Software | C++ | Debugging | Embedded SystemsSenior-level Full TimeMountain View, CA, USA11h ago
-
Software Engineer, Machine Learning USD 207K-300KC++ | Data Processing | Experimentation | Information Retrieval | Just-in-TimeSenior-level Full TimeNew York, NY, USA; Mountain View, …11h ago
-
Software Engineer III, AI/ML, Search News Intelligence USD 147K-211KAlgorithms | C++ | Data Processing | Data Structures | Feature EngineeringSenior-level Full TimeMountain View, CA, USA11h ago
-
Customer Engineer, Data Analytics, Google Cloud USD 153K-222KBatch Processing | Big Data | Cloud Architecture | Cloud platform | Customer RequirementsSenior-level Full TimeSunnyvale, CA, USA11h ago
-
C++ | Data Processing | Debugging | Information Retrieval | Language ModelsSenior-level Full TimeMountain View, CA, USA11h ago
-
Algorithms | C++ | Cloud Computing | Cloud platform | Data StructuresSenior-level Full TimeSunnyvale, CA, USA11h ago
-
Cloud Data and AI Engineer, Professional Services USD 127K-183KC++ | Capacity Planning | Cloud Databases | Data Migration | Data PipelinesTravel up to 30 percentMid-level Full TimeReston, VA, USA11h ago
-
Staff Software Engineer, ML Frameworks USD 207K-300KAPIs | Data Processing | Debugging | Fine Tuning | GPU AccelerationSenior-level Full TimeMountain View, CA, USA11h ago
-
Data Scientist USD 67K-150KA/B | A/B Testing | B testing | Clustering | Drift monitoring401k plan | AD and D insurance | Child Life Insurance | Dental insurance | Educational Assistance PlanMid-level Full TimeUnited States13h ago
-
Embedded Software Engineer 3 USD 90K-100KC++ | Code Analysis | Debugging | Development Lifecycle | Log Analysis401k matching | Annual bonus | Dental insurance | Disability coverage | Life insuranceSenior-level Full TimeChillicothe, IL15h ago
-
Chase Modeling - Applied AI ML Senior Associate USD 171K-215KClassification | Cloud Computing | Hadoop | Language Processing | Machine LearningBackup childcare | Financial coaching | Health care coverage | Mental health support | Onsite health and wellness centersSenior-level Full TimeChicago, IL, United States21h ago
-
AWS | Artificial Intelligence | Azure AI | Data Analysis | DatabricksBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimeChicago, IL, United States21h ago
-
Staff Machine Learning Engineer (Pricing) USD 215K-322KA/B Testing | API Design | AWS | Automated retraining | B testing401k matching | Dental insurance | Family planning assistance | Flexible time off | Healthcare benefitsSenior-level Full TimeSan Francisco, CA22h ago
-
Data Platform Engineer USD 205K-316KApache Airflow | Automation | Cloud infrastructure | Cloud platform | DBT401k match | Commuter benefits | Dental insurance | Employee stock options | Health insuranceMid-level Full TimeDenver, CO; New York, NY; San …22h ago
-
Anomaly Detection | Calibration | Classification | Clustering | Decision TreesCommuter benefits | Disability benefits | Health insurance | Life insurance | Paid time offSenior-level Full TimeNew York, New York22h ago
-
Agent systems | Anomaly Detection | Calibration | Classification | ClusteringDisability benefits | Equity awards | Health insurance | Life insurance | Paid time offSenior-level Full TimeBoston, Massachusetts22h ago