Staff Software Engineer, AI Engines 3P TPU Inference
Tasks
- Deliver ML infrastructure capability and optimization
- Guide sustainable engineering choices for ML systems at scale
- Innovate infrastructure directions
- Migrate TensorFlow JAX and PyTorch runtimes to ML Runtime
- Migrate custom workflows to ML Runtime
- Partner to transfer innovations into products
Perks/Benefits
- N/A
Skills/Tech-stack
Data Processing | Debugging | Distributed Computing | Fine Tuning | GPU | Information Retrieval | JAX | Language Processing | Large-Scale System Design | Large-scale | Large-scale system | ML Compilers | ML runtimes | Machine Learning | Machine Learning Infrastructure | Model Deployment | Model Evaluation | Natural Language | Natural Language Processing | PyTorch | System design | TPU | TensorFlow
Education
Roles
Regions
Countries
States
Related jobs
-
Featured Feat. Principal Knowledge & Data Architect USD 174K-284KAWS Neptune | Canonicalization | Chunking | Cypher | DBTBenefits including health and wellness programs | Health, wellness, and retirement plansSenior-level Full TimeHeadquarters - Chevy Chase, MD R22d ago
-
Research Engineer, Computer Vision USD 147K-208KAR/VR | Algorithm Development | Artificial Intelligence | C++ | Computer VisionSenior-level Full TimeSunnyvale, CA1h ago
-
C++ | Computer Vision | Concurrency | Data Structures | Data Structures and AlgorithmsBonuses | Equity | Health insurance | Paid time off | Parental leaveSenior-level Full TimeSan Jose, CA, USA2h ago
-
Senior Staff Software Engineer, AI/ML, Google Ads USD 262K-364KC++ | Data Processing | Data Structures | Data structures algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA2h ago
-
Applied Machine Learning | C++ | Concurrency | Distributed Systems | GoSenior-level Full TimeSan Jose, CA, USA2h ago
-
Staff Software Engineer, AI/ML GenAI, Google Cloud USD 207K-300KComputer Vision | Data Processing | Debugging | Distributed Systems | Fine TuningSenior-level Full TimeSunnyvale, CA, USA2h ago
-
C++ | Data Processing | Data Structures | Data Structures and Algorithms | Data durabilitySenior-level Full TimeKirkland, WA, USA2h ago
-
Customer Engineer III, Applied AI, Google Cloud USD 174K-252KAgent tooling | Application Performance Tuning | Application performance | C++ | CI/CDSenior-level Full TimeMountain View, CA, USA; New York, …2h ago
-
Software Engineer, Data Engineering USD 70K-200KAWS | AWS Glue | AWS Lambda | Amazon Athena | Amazon EC2Mid-level Full TimeIrvine, CA12h ago
-
Staff Frontier Agents Engineer (Applied AI) USD 252K-315KA/B | A/B Testing | AWS | Ablation Studies | AzureCommuter stipend | Generous PTO | Health, dental, vision coverage | Learning and development stipend | Retirement benefitsSenior-level Full TimeSan Francisco, CA; Seattle, WA; New …13h ago
-
Senior Frontier Agents Engineer (Applied AI) USD 216K-270KA/B Testing | AWS | Ablation Studies | Agent systems | AzureCommuter stipend | Health, dental, vision coverage | Learning and development stipend | Paid time off | Retirement benefitsSenior-level Full TimeSan Francisco, CA; Seattle, WA; New …13h ago
-
Frontier Agents Engineer (Applied AI) USD 180K-225KA/B | A/B Testing | AWS | Ablation Studies | Agent FrameworksCommuter stipend | Dental insurance | Health insurance | Learning and development stipend | Paid time offMid-level Full TimeSan Francisco, CA; Seattle, WA; New …13h ago
-
Senior AI Engineer - Services Special Projects USD 150K-260KAdversarial evaluation | Airflow | CI/CD | Cloud Platforms | Content ModerationSenior-level Full TimeSan Francisco Bay Area14h ago
-
Computer Vision Engineer USD 140K-260KC++ | CUDA | Cloud processing | Computer Vision | Deep learningSenior-level Full TimeSan Jose, California, United States15h ago
-
AI & Machine Learning Vice President USD 120K-220KAWS | Alteryx | Anomaly Detection | Azure | C#401k | Dental insurance | Disability insurance | Health insurance | Hybrid work modelExecutive-level Full TimeUnited States R15h ago
-
Software Engineer, ML Infra (Junior & New Grad) USD 125K-175KAWS | Azure | Bug tracking | C++ | Cloud ComputingEntry-level Full TimeMountain View, California, United States15h ago
-
Senior Embedded Software Engineer USD 191K-253KARM | Bare Metal | Board Bring-up | Bring-up | Build systemsComprehensive benefits | On-site work opportunities | U.S. Secret security clearance sponsorshipSenior-level Full TimeCosta Mesa, California, United States15h ago
-
Staff Machine Learning Engineer, Applied Research USD 189K-389KC++ | Graph Neural Networks | Information Retrieval | Java | Language ProcessingSenior-level Full TimeSan Francisco, CA, US; Remote, US R16h ago
-
Senior-level Full TimeUnited States16h ago
-
Machine Learning Engineer I, Network USD 151K-189KClassification | Data Pipelines | Deep learning | Experiment tracking | Feature Engineering401k match | Commuting support | Flexible PTO | Free lunch | Gym accessSenior-level Full TimeSan Francisco, CA16h ago
-
Consultant – Compliance Data Science & AI USD 98K-122KAPI | AWS | Anthropic | Azure | Dashboards401k match | Cell phone stipend | College savings | EAP | HSA & FSA optionsMid-level Full TimeSan Francisco, CA, United States16h ago
-
Allegro | BoM development | Bring-up | CE | CadenceDental insurance | Employer 401k match | Fitness programs | Learning and development | Medical insuranceSenior-level Full TimeScottsdale, Arizona, United States16h ago
-
Principal AI Engineer USD 185K-265KAPI Integration | Agent systems | Artificial Intelligence | Cloud Native | Cloud-native ArchitecturesEducation reimbursement | Health plans | Paid time off | Parental leave | Remote workSenior-level Full TimeRemote - USA R16h ago
-
Sr Machine Learning Engineer I USD 150K-241KAutomated retraining | Bayesian Filter | CI/CD | Cloud Computing | Data Association401k match | Dental insurance | Emotional & mental wellness support | Employee resource groups | Fitness programsSenior-level Full TimeSeattle, Washington, United States R16h ago
-
AI Engineer - FDE (Forward Deployed Engineer) USD 152K-210KAWS | Agent systems | Apache Spark | Azure | Databricks Intelligence PlatformRemote work | Travel opportunitiesSenior-level Full TimeUnited States16h ago