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
-
API Integration | Asynchronous processing | Containerization | Data Pipelines | Deep learningSenior-level ContractAnnapolis, United States R3h ago
-
AI Agents | C++ | Data Structures | Data Structures and Algorithms | Generative AISenior-level Full TimeNew York, NY, USA6h ago
-
Accelerated inference | Algorithms | Android | CPU | ChromeSenior-level Full TimeSunnyvale, CA, USA6h ago
-
ML Engineer, Query Understanding, Search Ads USD 174K-252KC++ | Data Processing | Data Storage | Debugging | Deep learningMid-level Full TimeMountain View, CA, USA6h ago
-
BigQuery | Cloud platform | Data Processing | Deep learning | Distributed SystemsMid-level Full TimeKirkland, WA, USA6h ago
-
Software Engineer, Machine Management Qual Infra USD 147K-211KArtificial Intelligence | C++ | Fleet Management | Machine Learning | Software developmentMid-level Full TimeSunnyvale, CA, USA6h 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, …6h 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, USA6h ago
-
Data Structures | Data Structures and Algorithms | Deep learning | Diffusion Models | Generative AISenior-level Full TimeMountain View, CA, USA6h 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, USA6h 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, USA6h ago
-
Machine Learning Engineer, XLS, Compiler USD 174K-252KAI | C++ | Compiler Construction | Data Processing | Data StructuresMid-level Full TimeChicago, IL, USA6h ago
-
Senior ML/AI Software Engineer, Chrome AI USD 174K-252KAgent Orchestration | Algorithms | Automated testing | C++ | Computer VisionSenior-level Full TimeMountain View, CA, USA6h ago
-
Staff Software Engineer, AI/ML, Agent Assist USD 207K-300KAgentic AI | Cloud Computing | Data Processing | Debugging | Deep learningSenior-level Full TimeSunnyvale, CA, USA6h 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, 56007611h ago
-
Machine Learning Engineer, API Multicloud USD 295K-445KAWS | Data Pipelines | Data Structures | Data structures algorithms | Deep learningSenior-level Full TimeSan Francisco15h ago
-
Agentic AI | Agile | Cloud Computing | Container Orchestration | Distributed SystemsSenior-level Full TimeUnited States16h 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, CA16h ago
-
Tech Lead, Robotic AI Model USD 150K-180KAction Chunking | Action Tokenization | Behavior Cloning | DPO | DeepSpeedSenior-level Full TimeEl Segundo, California, United States16h 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 Francisco17h 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) R17h ago
-
Data analytics | Language Processing | Machine Learning | Natural Language | Natural Language ProcessingCollaborative environment | Mentorship | TrainingSenior-level Full TimeSan Francisco, California, United States - … R17h ago
-
Senior Technology Engineer- GenAI USD 65K-171KArtificial Intelligence | Development Life Cycle | Generative AI | Machine Learning | Model EvaluationSenior-level Full TimeTwo PNC Plaza (PA374), United States17h 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 States17h 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, CA18h ago