Software Engineer, Performance, YouTube ML Efficiency
Tasks
- Act as TPU hardware expert
- Address infrastructure and compiler tuning issues
- Design self-tuning frameworks and analytics
- Develop new optimizers
- Identify and unblock performance issues
- Implement ragged tensors
- Partner with quality teams for model changes
- Push hardware limits with SparseCore
- Redesign components for efficiency
- Scale embedding tables
- Write low-level Pallas kernels
Perks/Benefits
Skills/Tech-stack
Algorithms | Artificial Intelligence | Computer Architecture | Data Analysis | Data Storage | Data Structures | Debugging | Distributed Computing | GPU | JAX | Language Processing | ML Compilers | ML frameworks | Mobile development | Natural Language | Natural Language Processing | Networking | Performance Analysis | PyTorch | Security | System design | Systems data analysis | TPU | TensorFlow | UI Design | Visualization Tools | XLA
Education
Regions
Countries
States
Cities
Related jobs
-
Software Engineer III, AI/ML, Google Ads USD 147K-211KC++ | Data Processing | Data Storage | Debugging | Distributed ComputingSenior-level Full TimeMountain View, CA, USA; Los Angeles, …4h ago
-
Senior Software Engineer, AI/ML, AI and Infrastructure USD 174K-252KC++ | Data Processing | Debugging | Language Processing | ML InfrastructureSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Software Engineer III, AI/ML, Google Workspace USD 147K-211KC++ | Data Processing | Debugging | Language Processing | ML InfrastructureSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Software Engineer III, AI/ML GenAI, Google Cloud Compute USD 147K-211KAudio generation | C++ | Cloud platform | Data Preparation | Data ProcessingSenior-level Full TimeKirkland, WA, USA4h ago
-
Senior Software Engineer, AI/ML, Google Cloud Compute USD 174K-252KC++ | Data Processing | Debugging | Distributed Computing | Language ProcessingSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Senior Software Engineer, AI/ML GenAI, Search USD 174K-252KAlgorithms | C++ | Code review | Computer Vision | Data ProcessingSenior-level Full TimeMountain View, CA, USA4h ago
-
Senior Software Engineer, AI/ML GenAI, Google Play USD 174K-252KC++ | Computer Vision | Data Processing | Data Storage | DebuggingSenior-level Full TimeMountain View, CA, USA4h ago
-
Software Engineer, BigQuery AI/ML USD 147K-211KArtificial Intelligence | BigQuery | C plus plus | Cloud platform | Compute TechnologiesMid-level Full TimeKirkland, WA, USA4h ago
-
Software Engineer III, AI/ML, AI and Infrastructure USD 147K-211KAlgorithms | C++ | Data Processing | Data Structures | DebuggingSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Sr Lead Software Engineer - AI/ML USD 186K-215KApplication development | Artificial Intelligence | Cloud Native | Code review | DebuggingBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimePlano, TX, United States10h ago
-
Staff Software Engineer - Distributed Data Systems USD 182K-247KACID transactions | Algorithms | Apache Spark | C++ | Data StructuresSenior-level Full TimeBellevue, Washington16h ago
-
Senior Software Engineer - Distributed Data Systems USD 157K-213KACID transactions | Algorithms | Apache Spark | C++ | Data StructuresSenior-level Full TimeBellevue, Washington16h ago
-
Senior AI Software Engineer USD 160K-198KA2A | ADK | API | Containers | EnvoyNo visa sponsorship | Remote workSenior-level Full TimeRemote US R17h ago
-
AI & ML Engineering Intern USD 104K-166KAutomation | DBT | Data Connectors | Embeddings | EvaluationEquity | Flexible schedule | Free snacks | Paid time off | Ping-pong tableEntry-level InternshipAustin, Texas18h ago
-
Systems Software Engineer - Robotic Platform USD 150K-200K*nix | C++ | Deterministic builds | Docker | LinuxSenior-level Full TimeAustin, Texas, United States18h ago
-
Staff Machine Learning SWE Infra USD 238K-302KAutoregressive models | Cloud Computing | Data Engineering | Distributed Training | Gradient ShardingCompany benefits program | Discretionary annual bonus | Equity incentive planSenior-level Full TimeMountain View, CA, USA18h ago
-
Senior Developer, Data & IT - AI Solutions USD 120K-142KAI Agents | API Integration | AWS | AWS Bedrock | AWS SageMakerDental insurance | Dependent Care Account | Health insurance | Health savings account | Mental health counseling supportSenior-level Full TimeNew York, NY, United States18h ago
-
Member of Technical Staff - Hardcore Supercompute USD 180K-440KAnsible | C plus plus | Containerization | Debugging | Distributed Systems401k retirement plan | Dental insurance | Disability insurance | Employee discounts | Life insuranceSenior-level Full TimePalo Alto, CA; Seattle, WA19h ago
-
Distinguished Software Engineer, Data Infrastructure USD 248K-406KAI Inference | AI Training | Batch Processing | Compliance | Data InfrastructureExecutive-level Full TimeMountain View, CA, United States21h ago
-
APIs | Agile | Code review | Data Analysis | Machine LearningMentorship | Remote workEntry-level Apprenticeship ContractRemote (United States) R21h ago
-
Agile | Code review | Databases | Debugging | JavaMentorship | Remote work | Tuition coverageEntry-level Apprenticeship ContractRemote (United States) R21h ago
-
Lead Software Engineer - Cloud/AI Engineer USD 178K-215KAI Agents | AWS | Algorithms | CI/CD | Data StructuresBackup childcare | Financial coaching | Health care coverage | Mental health support | Retirement savings planSenior-level Full TimePlano, TX, United States22h ago
-
AI/ML Lead Software Engineer USD 185K-215KA/B | A/B Testing | Agile | B testing | CI/CDBackup childcare | Financial coaching | Health care coverage | Mental health support | Retirement savings planSenior-level Full TimeNew York, NY, United States23h ago
-
Software Engineer III (SWES03) DT_AI/ML USD 165K-190KAI Platform | AWS SageMaker | Airflow | Azure ML | CI/CDBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellnessSenior-level Full TimeNew York, NY, United States1d ago
-
Software Engineer - Language (Technical Leadership) USD 213K-293KASR | Benchmarking | C# | C++ | Conversational AISenior-level Full TimeMenlo Park, CA | Seattle, WA …1d ago