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
-
Senior Product Engineer, Machine Learning Accelerators USD 144K-209KBash | Bench testing | Board assembly | Bring-up | Cause analysisSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Software Engineer, Storage Everywhere, Google Cloud USD 147K-211KBack-end | Back-end development | C++ | Cloud Platforms | Data StructuresMid-level Full TimeSunnyvale, CA, USA3h ago
-
Senior Staff Software Engineer, Torch TPU USD 262K-365KAlgorithms | C++ | Cloud Computing | Computer Vision | Data ProcessingSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Staff Software Engineer, Core ML, Frameworks USD 207K-300KAlgorithms | Audio Processing | Cross Business Projects | Cross-Functional Collaboration | Cross-functionalSenior-level Full TimeSunnyvale, CA, USA3h 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, USA3h ago
-
Senior Software Engineer, AI/ML GenAI, Google Play USD 174K-252KAndroid | C++ | Computer Vision | Data Processing | Data StorageSenior-level Full TimeMountain View, CA, USA3h 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, …3h 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, …3h ago
-
RTL Design Engineer, Machine Learning Accelerators USD 138K-198KASIC design | Code review | High Performance | High-Performance Design | Low powerMid-level Full TimeSunnyvale, CA, USA3h ago
-
C# | C++ | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA3h ago
-
Senior Software Engineer, AI/ML, AIR USD 174K-252KAlgorithms | Android | Data Structures | Evaluation Pipeline | GenAI agentsSenior-level Full TimeMountain View, CA, USA3h 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, …3h 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, …3h ago
-
Data Processing | Data Structures | Data Structures and Algorithms | Debugging | Fine TuningSenior-level Full TimeSunnyvale, CA, USA; New York, NY, …3h 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 Francisco11h 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 States14h 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 States15h 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, …15h 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 States16h 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 States17h 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, London17h ago
-
Sr. Staff Embedded AI Engineer USD 165K-200KBare Metal | C# | C++ | CMSIS NN | Code generationEmployee resource groups | Flexible work environment | Hybrid schedule | Remote work optionSenior-level Full TimeColumbia, MARYLAND, United States17h ago
-
Senior Software Engineer, Robotics USD 141K-194KAlgorithms | C++ | CI/CD | Computer Vision | Constraint SatisfactionCompany holidays | Health insurance | Learning reimbursement | Life insurance | Long-term disabilitySenior-level Full TimeTorrance, California, United States17h ago
-
Principal Machine Learning Engineer USD 157K-223KA/B | A/B Testing | AI Agents | Airflow | B testingSenior-level Full TimeUS TX Austin17h ago
-
Senior Forward Deployed Engineer (AI/ML) USD 140K-174KBenchmarking | CUDA | Continuous batching | CrewAI | DatabasesConference reimbursement | Employee assistance program | Flexible time off | LinkedIn Learning access | Local Employee MeetupsSenior-level Full TimeSeattle R18h ago