Staff Software Engineer, TPU Performance
Tasks
- Achieve performance for customer launches
- Analyze performance and efficiency metrics
- Design and implement fleet scale performance solutions
- Identify performance opportunities using benchmarks
- Improve compiler and runtime performance
- Maintain ML training and serving benchmarks
- Onboard ML models and products on TPU hardware
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | Code generation | Compiler optimization | Data Processing | Data Structures | Data Structures and Algorithms | Fine Tuning | GPU Programming | JAX | MLIR | Machine Learning | Machine Learning Debugging | Model Deployment | Model Evaluation | Model Fine-tuning | OpenCL | OpenXLA | Product Launch | PyTorch | Reinforcement Learning | Runtime Systems | Software Architecture | Software product launch | Software testing | TPU Performance | Tensor Processing Units | Tensor processing | Triton
Education
Roles
Regions
Countries
States
Related jobs
-
Sr. Data Scientist USD 90K-132KClustering | Component analysis | Dashboards | Data Quality | Data Visualization401k matching | Dental insurance | Disability insurance | Educational growth | Employee discount programSenior-level Full TimeTexas-The Woodlands2h ago
-
Automation | Distributed Systems | Linux | Machine Learning | Performance AnalysisSenior-level Full TimeSan Jose, California, United States3h ago
-
Software Engineer, Ads Machine Learning Data Platform USD 212K-389KData Engineering | Feature Engineering | Feature Store | Machine Learning | PrivacySenior-level Full TimeSan Jose, California, United States3h ago
-
Senior Product Engineer, Machine Learning Accelerators USD 144K-209KBash | Bench testing | Board assembly | Bring-up | Cause analysisSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Software Engineer, Storage Everywhere, Google Cloud USD 147K-211KBack-end | Back-end development | C++ | Cloud Platforms | Data StructuresMid-level Full TimeSunnyvale, CA, USA4h ago
-
Senior Staff Software Engineer, Torch TPU USD 262K-365KAlgorithms | C++ | Cloud Computing | Computer Vision | Data ProcessingSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Staff Software Engineer, Core ML, Frameworks USD 207K-300KAlgorithms | Audio Processing | Cross Business Projects | Cross-Functional Collaboration | Cross-functionalSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Senior Software Engineer, AI/ML GenAI, Google Cloud USD 174K-252KC plus plus | Computer Vision | Data Processing | Debugging | Distributed ComputingSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Senior Software Engineer, AI/ML GenAI, Google Play USD 174K-252KAndroid | C++ | Computer Vision | Data Processing | Data StorageSenior-level Full TimeMountain View, CA, USA4h ago
-
Staff Software Engineer, AI/ML GenAI, Google Cloud USD 207K-300KAlgorithms | C++ | Cloud infrastructure | Computer Vision | Data ProcessingSenior-level Full TimeSunnyvale, CA, USA; San Francisco, CA, …4h ago
-
Senior Software Engineer, AI/ML, Google Workspace USD 174K-252KC++ | Data Processing | Debugging | Language Processing | ML InfrastructureSenior-level Full TimeNew York, NY, USA; Sunnyvale, CA, …4h ago
-
RTL Design Engineer, Machine Learning Accelerators USD 138K-198KASIC design | Code review | High Performance | High-Performance Design | Low powerMid-level Full TimeSunnyvale, CA, USA4h ago
-
C# | C++ | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA4h ago
-
Senior Software Engineer, AI/ML, AIR USD 174K-252KAlgorithms | Android | Data Structures | Evaluation Pipeline | GenAI agentsSenior-level Full TimeMountain View, CA, USA4h ago
-
Software Engineer III, AI/ML, Google Fi USD 147K-211KData Processing | Debugging | Distributed Systems | Java | Language ProcessingSenior-level Full TimeMountain View, CA, USA; San Francisco, …4h ago
-
Senior Software Developer, Embedded Systems/Firmware, XR USD 174K-252KAlgorithms | Android | Android HAL | Bootloader | C#Senior-level Full TimeSan Jose, CA, USA; Waterloo, ON, …4h ago
-
Data Processing | Data Structures | Data Structures and Algorithms | Debugging | Fine TuningSenior-level Full TimeSunnyvale, CA, USA; New York, NY, …4h ago
-
Vertical AI Lead USD 150K-250KAPI Integration | Agentic Systems | Artificial Intelligence | Customer discovery | Go-to-marketBenefits | Equity | Visa sponsorshipSenior-level Full TimeNew York City / San Francisco12h ago
-
Applied AI/ML - Senior Associate USD 180K-215KArtificial Intelligence | Data Analysis | Machine Learning | Model Evaluation | PyTorchBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimeChicago, IL, United States13h ago
-
AI Solution Principal Systems Development Engineer USD 148K-222KCluster management | Containerization | GPU Computing | Generative AI | Hardware/Software Integration401k plan | Choice of medical coverage | Fitness reimbursement | Generous Time Off Programs | Team member discounts on Dell productsSenior-level Full TimeAustin, Texas, United States, United States15h ago
-
Junior Applied AI Engineer- HRIS USD 105K-140KAWS | Azure | Cloud Platforms | Cloud platform | Data Engineering401k | Dental insurance | Disability insurance | Leave Program | Life insuranceEntry-level Full TimeSunnyvale, CA, United States16h ago
-
API Security | AWS | Adversarial inputs | Agent Frameworks | Application SecurityCiti Bike membership | Flexible paid time off | Gym membership | Health and mental health services | Home office stipendSenior-level Full TimeSan Francisco, California, United States; Seattle, …16h ago
-
Applied AI ML, Senior Associate USD 171K-245KAgentic AI | Cloud services | Continuous integration | Data Analysis | Data PipelinesBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimeHouston, TX, United States18h ago
-
Agentic AI | Data Analysis | Data Pipelines | Financial Data | Financial Data AnalysisBackup childcare | Financial coaching | Health care coverage | Mental health support | On Site Health Wellness CentersEntry-level Full TimeJersey City, NJ, United States18h ago
-
Applied AI Engineer (AI Adoption) USD 200K-250KAI Evaluation | Agent Design | Automation | LLM APIs | Machine LearningDental insurance | Group pension scheme | Income protection | Life assurance | Paid HolidaysMid-level Full TimeNew York, Chicago, London18h ago