Senior Staff Software Engineer, TPU Performance
Tasks
- Achieve performance for customer launches
- Co design TPU friendly models
- Collaborate with partner teams
- Drive performance optimizations
- Exploit TPU hardware features
- Identify ML training and serving benchmarks
- Maintain ML training and serving benchmarks
- Participate in algorithmic innovations
- Participate in model preserving optimizations
- Use benchmarks to find performance opportunities
Perks/Benefits
Skills/Tech-stack
Auto sharding | Benchmarking | C++ | Compiler optimization | Computer Engineering | Data Processing | Data Structures | Data Structures and Algorithms | Debugging | JAX | LoRA | Machine Learning | Model Deployment | Model Evaluation | Performance Analysis | Product Architecture | PyTorch | Python | Quantization | Reinforcement Learning | Runtime optimization | Software Product Architecture | Software testing | Sparsity | Speculative decoding | Speech Processing
Education
Regions
Countries
States
Cities
Related jobs
-
AI and Data Science Engineer (TS/SCI Poly) USD 93K-163KAPIs | Artificial Intelligence | CI/CD | Cloud Platforms | ContainerizationMid-level Full TimeMcLean, Virginia, United States1h ago
-
Senior AI/ML Engineer USD 117K-195KAPI Development | AWS | AWS SageMaker | Airflow | AzureCareer growth | Customer-focused work | Onsite Pentagon work | Rapid advancementSenior-level Full TimeUSA-VA-Arlington1h ago
-
Mid-level Full TimeUSA-VA-Arlington1h ago
-
AI Forward Deployment Engineer USD 130K-145KAPIs | AWS | Airflow | Artificial Intelligence | AzureClient site travel | On-site workEntry-level Full TimeNew York, New York, United States1h ago
-
Staff Software Engineer, Generative AI USD 207K-300KAndroid | C++ | Computer Vision | Data Processing | DebuggingSenior-level Full TimeSunnyvale, CA, USA3h ago
-
RTL Design Engineer, Machine Learning Accelerators USD 138K-198KASIC | Code review | High Performance | High-Performance Design | Learning algorithmsMid-level Full TimeSunnyvale, CA, USA3h ago
-
C++ | Computer Graphics | Data Curation | Data Pipelines | Human-in-the-loopSenior-level Full TimeNew York, NY, USA; Mountain View, …3h ago
-
Senior Software Engineer, Managed Spark, OpenSource USD 174K-252KApache Flink | Apache Hive | Apache Hudi | Apache Iceberg | Apache SparkSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Software Engineer III, AI/ML Machine Learning, Core USD 147K-211KC++ | Compute Technologies | Data Preprocessing | Data Storage | Distributed SystemsSenior-level Full TimeNew York, NY, USA; Mountain View, …3h ago
-
Research Scientist, Google Research, GenAI, Experiences USD 147K-211KComputer Vision | Data Mining | Deep learning | High Performance | High-Performance ComputingMid-level Full TimeMountain View, CA, USA; Kirkland, WA, …3h ago
-
Staff Software Engineer, Storage USD 207K-301KC# | C++ | Data Storage | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA3h ago
-
Software Engineer III, Embedded Systems/Firmware, XR USD 147K-211KAlgorithms | Android | Android HAL | Bootloader | C#Senior-level Full TimeMiami, FL, USA; San Jose, CA, …3h ago
-
Software Engineer, AI/ML Computer Vision USD 147K-211KC++ | Computer Vision | Data Processing | Data Structures | Data Structures and AlgorithmsMid-level Full TimeSeattle, WA, USA; Kirkland, WA, USA3h ago
-
Senior Software Developer, Embedded Systems/Firmware, XR USD 174K-253KC++ | Code review | Computational Photography | Computer Vision | Data StructuresSenior-level Full TimeSan Jose, CA, USA; Miami, FL, …3h ago
-
Software Engineer, ML Quality, Google Labs USD 207K-300KA/B | A/B Testing | Algorithms | Artificial Intelligence | B testingBonus and equity | Career development | Health insurance | Paid time off | Relocation assistanceSenior-level Full TimeMountain View, CA, USA; San Francisco, …3h ago
-
Sr. Data Scientist USD 114K-220KAirflow | Anomaly Detection | Cloud Platforms | Convolutional Neural Networks | Data acquisition401k matching | Employee stock purchase plan | Healthcare benefits | Paid time off | Parental leaveSenior-level Full TimeIrvine, CA, US5h ago
-
Research Engineer USD 180K-300KData Curation | Distributed Training | Foundation Models | ML Performance Optimization | ML performance401k match | Daily lunches and snacks | Health benefits | Learning and development stipend | Relocation assistanceMid-level Full TimeRedwood City8h ago
-
Senior ML/AI Modeler, Risk Automation Machine Learning USD 160K-283KAPI Integration | AWS | Airflow | Artificial Intelligence | Batch ProcessingFlexible time off | Medical insurance | Modern family planning | Remote work | Retirement savings plansSenior-level Full TimeSeattle, WA, United States of America R8h ago
-
Autonomy | C++ | DDS | Edge Computing | Embedded Systems401k | Health insurance | Paid time offMid-level Full TimeSan Carlos - Hybrid R12h ago
-
Autonomy | C++ | DDS | Edge Computing | Embedded Systems401k plan | Health insurance | Paid Company Holidays | Paid time off | Phone stipendSenior-level Full TimeSan Carlos - Hybrid R12h ago
-
Machine Learning Engineer III, Core Agents USD 175K-219KAWS Bedrock | Embeddings | Evaluation metrics | Indexing | Information RetrievalEmployee benefits | Equity compensation | Flexible work schedule | In office collaboration at least 3 days per weekSenior-level Full TimeRedwood City, CA13h ago
-
Sr. Firmware Engineer, Annapurna Labs, Machine Learning Acceleration - Power and Performance USD 193K-261KData Pipelines | Embedded C | Firmware | Frequency Voltage | Frequency Voltage ScalingCode reviews | Knowledge sharing | MentorshipSenior-level Full TimeCupertino, California, USA14h ago
-
Senior Staff Machine Learning Engineer USD 242K-357KA/B | A/B Testing | B testing | C++ | Deep learning11 paid holidays | 401k employer matching | Basic life insurance | Commuter benefits | Dental insuranceSenior-level Full TimeSan Francisco, CA; Sunnyvale, CA15h ago
-
Sr. Embedded Security Engineer USD 138K-184KAnti-rollback | Applied cryptography | Authenticated Boot | Authentication | Bare MetalEmployee resource groups | Flexible hours | Flexible time off | Medical, dental & vision coverage | Monthly social eventsSenior-level Full TimeSan Francisco15h ago
-
Staff Data Engineer, Ads USD 248K-279KAPI Integration | Airflow | Alerting | Anomaly Detection | BigQuerySenior-level Full TimeRemote (U.S.) R15h ago