Staff Software Engineer, TPU Performance
Sunnyvale, CA, USA; Kirkland, WA, USA
USD 207K-300K Senior-level Full Time
Tasks
- Analyze performance and efficiency metrics
- Explore model and data efficiency techniques
- Identify bottlenecks and implement fleet scale solutions
- Identify performance opportunities using benchmarks
- Improve compiler and runtime performance
- Onboard new ML models and products to TPU hardware
- Optimize ML training and serving performance
- Perform TPU fleet efficiency analysis
Perks/Benefits
- N/A
Skills/Tech-stack
Benchmarking | Compiler optimization | Data Analysis | Data Structures | Data Structures and Algorithms | Data Visualization | Debugging | Distributed Systems | JAX | Machine Learning | Performance Tuning | PyTorch | Quantization | Runtime optimization | Software Architecture | Software Deployment | Software testing | Sparsity | Tensor Processing Unit | Tensor processing
Education
Roles
Regions
Countries
States
Related jobs
-
A/AI Research Engineer Asc - E1 USD 67K-124KC++ | CIN? N/A | Computer Vision | Control Engineering | Control Systems401k match | Dental insurance | Education assistance | Flexible work arrangements | Life insuranceEntry-level Full TimeOrlando, FL, US6h ago
-
Embedded Software Engineering Asc Mgr USD 120K-224KAgile | Algorithm Implementation | C++ | Confluence | Critical softwareFlexible work schedule | Work-life balanceSenior-level Full TimeOrlando, FL, US6h ago
-
Algorithm Engineer, Tracking Systems USD 126K-167KAgile | C++ | Clustering | Data Association | Differential EquationsMid-level Full TimeBroomfield, Colorado, United States; Fort Collins, …7h ago
-
Algorithm Developer, Tracking Systems USD 126K-167KAgile | C++ | Clustering | Computer Vision | Data AssociationHealth benefits | Recovery supportMid-level Full TimeBroomfield, Colorado, United States; Fort Collins, …7h ago
-
Senior Algorithm Engineer, Tracking Systems USD 190K-252KAgile Development | C++ | Clustering | Controls | Data AssociationHealth insurance | Performance bonusesSenior-level Full TimeBroomfield, Colorado, United States; Fort Collins, …8h ago
-
ML Infra Engineer, Modeling USD 200K-270KCheckpointing | Cloud Computing | Data loading | Distributed Training | Experiment trackingSenior-level Full TimeSan Francisco10h ago
-
API Integration | Artificial Intelligence | Data Engineering | Data Pipelines | Data provenanceSenior-level Full TimeUS026 PA Spring House - 1400 …11h ago
-
Machine Learning Engineer, Relevance and Personalization USD 166K-195KA/B | A/B Testing | Apache Airflow | Apache Kafka | Apache SparkMid-level Full TimeUnited States R11h ago
-
Data Platform Software Engineer USD 125K-194KCI/CD | Cloud Computing | Data Lakes | Data Modeling | Data Structures401k match | ESPP | Family leave programs | Flexible time off | Health plansEntry-level Full TimeSan Diego, CALIFORNIA, United States R11h ago
-
Senior Machine Learning (ML) Engineer USD 175K-200KAI Search | Artificial Intelligence | Attention Mechanisms | Autogen | Azure AI401k plan | Dental insurance | Flexible paid time off | Flexible work hours | Medical insuranceSenior-level Full TimeSeattle, WA R13h ago
-
Software Engineer, Data Runtime USD 29KAWS | Admission Policy | Arrow | Azure | Benchmarking401k | Additional company holidays | Enhanced parental leave | Generous vacation policy | Holiday breakSenior-level Full TimeSan Francisco, US (Hybrid) R13h ago
-
Apache Iceberg | Apache Spark | Avro | Concurrency | Data Ingestion401k matching | ESPP | Family leave programs | Flexible spending accounts | Flexible time awaySenior-level Full TimeSan Diego, CALIFORNIA, United States R13h ago
-
AI Engineer I USD 90K-119KAPI Integration | Azure | CI/CD | Docker | Embeddings401k match | Dental insurance | Holiday pay | Medical insurance | Paid time offEntry-level Full TimeRemote, CA, USA R13h ago
-
API | AWS | Automated testing | Cloud Computing | Code generation401k | Dental insurance | Discounts and perks | Fully remote | Medical insuranceSenior-level Full TimeUniversal City, CALIFORNIA, United States R14h ago
-
Senior Software Engineer (Data Platform) USD 185K-185K.NET | AWS | Authentication | Authorization Code | Authorization Code GrantSenior-level Full TimeUnited States (EST) R14h ago
-
Software Engineer II - Data Security USD 149K-214KAudit Logging | Authentication | Authorization | CAP Theorem | Cloud infrastructureCross-functional collaboration | Leadership opportunities | Remote-first work environmentMid-level Full TimeRemote - USA R16h ago
-
Applied AI Research (Capabilities) USD 300K-405KDeep learning | Evals | Evaluation | Finetuning | Language ModelsCompetitive benefits | Flexible working hours | Generous vacation | Optional equity donation matching | Parental leaveMid-level Full TimeSan Francisco, CA | New York … R17h ago
-
AI Model Engineer USD 125K-150KAWS Glue | Agentic Workflow | Apache Airflow | Apache Spark | BashHybrid Remote Onsite | Public Trust requiredSenior-level Full TimeFAIRFAX, VA, United States18h ago
-
Staff Software Engineer - Snowflake Feature Store USD 236K-339KComputer Science | Computer science fundamentals | Data Infrastructure | Feature Engineering | JavaSenior-level Full TimeUS-WA-Bellevue18h ago
-
Senior Software Engineer - ManagedDB USD 200K-287KAWS | Ansible | Azure | C plus plus | Cloud NetworkingSenior-level Full TimeUS-CA-Menlo Park R20h ago
-
Artificial Intelligence | C plus plus | C# | Embedded Systems | Firmware DevelopmentEntry-level Full TimeMoon Township, Pennsylvania, USA, 1510820h ago
-
Lead AI Engineer USD 125K-180KAPIs | AWS | AWS Bedrock | Agentic Workflows | Azure401k | Commuter benefits | Dental insurance | Disability coverage | Employee assistance programSenior-level Full TimeColumbia, MD, United States20h ago
-
Research Engineer, Agentic Systems USD 175K-275KAgentic Systems | Evaluation | Experimentation | Fine Tuning | Language Models401k | Catered lunch | Commuter benefits | Dental insurance | Dinner stipendEntry-level Full TimeUnion Square, New York City20h ago
-
Senior-level Full TimeGainesville, FL20h ago
-
Lead Software Engineer USD 140K-185KAI Foundry | Access Management | Azure | Azure AI | Azure AI FoundrySenior-level Full TimeGainesville, FL20h ago