Software Engineer, TPU Inference, AI/ML
Tasks
- Analyze performance on TPUs and GPUs
- Ensure performance on heterogeneous hardware
- Implement LLM and recommendation models
- Implement diffusion model architectures
- Implement models in JAX and PyTorch
- Leverage TPU hardware features
- Optimize TPU execution
- Research model architectures
- Transition recommendation models to TPUs
- Verify model correctness
Perks/Benefits
- N/A
Skills/Tech-stack
Cloud Computing | Compilers | GPU | GPU Programming | Inference | JAX | JIT Compilation | Kubernetes | Language Models | Language Processing | Large Language Models | Machine Learning | Model Optimization | Natural Language | Natural Language Processing | Performance Tuning | PyTorch | Python | TPUs
Education
Roles
Related jobs
-
AI Research Engineer USD 177K-251KBenchmarks | Data Pipelines | Data Versioning | Evaluation | Fine TuningCross-functional collaboration | End-to-end ownership | High autonomyMid-level Full TimeBellevue, WA | Menlo Park, CA …2h ago
-
Data Engineer USD 185K-196KApache Spark | Artificial Intelligence | CSS | Data Governance | Data ModelingMid-level Full TimeMenlo Park, CA2h ago
-
Operations Platform Engineer USD 153K-200KAPI Design | Alerting | Anomaly Detection | Backpressure | BufferingSenior-level Full TimeRedmond, WA2h ago
-
Entry-level Full TimeBellevue, WA2h ago
-
Entry-level Full TimeMenlo Park, CA2h ago
-
Privacy Engineer USD 194K-217KApache Airflow | Apache Spark | Automated testing | C plus plus | Continuous DeploymentEntry-level Full TimeMenlo Park, CA2h ago
-
Software Engineer III, AI/ML GenAI, Google Cloud Compute USD 147K-211KAudio generation | C++ | Computer Vision | Data Processing | Data StorageSenior-level Full TimeKirkland, WA, USA2h ago
-
Senior Software Engineer, AI/ML, Google Cloud AI USD 174K-252KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Software Engineer, Machine Health USD 147K-211KAnalysis and Design | C++ | Data Processing | Data analytics | Distributed SystemsMid-level Full TimeSunnyvale, CA, USA2h ago
-
Networking AI Technical Lead USD 207K-300KAI | Algorithms | C++ | Compute Technologies | Data StructuresSenior-level Full TimeSunnyvale, CA, USA; Cambridge, MA, USA2h ago
-
Software Engineer, Decoding, Quantum AI USD 147K-211KC++ | Concurrency | DSP | Error correction | FPGAMid-level Full TimeLos Angeles, CA, USA; Goleta, CA, …2h ago
-
Software Engineer III, AI/ML, Health and Home USD 147K-211KData Analysis | Data Processing | Data Storage | Debugging | Distributed ComputingSenior-level Full TimeMountain View, CA, USA2h ago
-
Customer Engineer II, Applied AI, Google Cloud USD 149K-216KApplication Architecture | Artificial Intelligence | C++ | Cloud Computing | Cloud platformSenior-level Full TimeNew York, NY, USA; Chicago, IL, …2h ago
-
Software Engineer III, AI/ML GenAI, Google Cloud AI USD 147K-211KC++ | Data Processing | Debugging | Generative AI | Language ModelsSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Senior Software Engineer, Full Stack, Gen AI Formats USD 174K-252KArtificial Intelligence | C++ | Data Storage | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeMountain View, CA, USA2h ago
-
Staff Software Engineer, TPU Performance USD 207K-300KAlgorithm Design | Benchmarking | Compiler | Data Analysis | Data StructuresSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Customer Engineer, Data Analytics, Google Cloud USD 153K-222KBatch Processing | Big Data | Cloud Native | Cloud Native Architecture | Cloud platformSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Senior Software Engineer, AI/ML, Retail Ads, Fullstack USD 174K-252KAlgorithms | C++ | Data Processing | Data Structures | DebuggingSenior-level Full TimeMountain View, CA, USA2h ago
-
Algorithms | Data Structures | Deep learning | Embeddings | Information RetrievalSenior-level Full TimeMountain View, CA, USA2h ago
-
Senior Software Engineer, AI Infra, Ads Safety USD 174K-252KBenchmarking | C++ | Data Processing | Data Storage | Data platformSenior-level Full TimeLos Angeles, CA, USA2h ago
-
AWS | Azure | CI/CD | Data Engineering | DockerCross-functional collaboration | Documentation support | Flexible teamwork environment | MentorshipEntry-level InternshipIrvine, CA, US5h ago
-
AI Engineer USD 109K-155KAI Agents | API Integration | Generative AI | Language Models | Large Language Models401k matching | Dental insurance | Disability coverage | Medical insurance | Paid HolidaysSenior-level Full TimeOverland Park, KS, US6h ago
-
Senior Software Engineer, Data Platform & AI Enablement USD 158K-260KAI infrastructure | Checkpointing | Distributed Systems | Exactly once | Exactly-once semanticsSenior-level Full TimeUS - San Francisco7h ago
-
Staff Backend Software Engineer- (AI Platform) USD 166K-225KAPI Design | Algorithms | Autoscaling | Caching | Data StructuresSenior-level Full TimeSan Francisco, California9h ago
-
Staff Backend Software Engineer- (AI Platform) USD 192K-260KAPI Design | APIs | Algorithms | Autoscaling | CachingSenior-level Full TimeSan Francisco, California9h ago