44 jobs for TPU

Senior Software Engineer, AI/ML, NotebookLM Content Studio USD 174K-253K

C++ | Debugging | GPU | Java | LLM

Senior-level Full Time

New York, NY, USA; Cambridge, MA, …

1h ago

AI Engineer - Mission Innovation Lab USD 125K-188K

AI Safety | Adversarial Testing | Airflow | C# | C++

Childcare support | Comprehensive medical insurance | Employer Retirement Contributions | Fitness center access | Free Transit Bus Pass

Senior-level Full Time

Pittsburgh, United States

1d ago

ML Infra Engineer (TPU/Jax/Optimization) USD 197K-291K

AWS | Checkpointing | Cloud Computing | Data loading | Device Utilization

Senior-level Full Time

San Francisco

2d ago

Software Engineer III, GenAI Data Operations Research, XR GBP 225K-325K

Data Filtering | Data Processing | Data debugging | Deep learning | Distributed Computing

Senior-level Full Time

London, UK

3d ago

Senior Software Engineer, AI/ML, Ads Training USD 174K-253K

C++ | Distributed Systems | GPU | JAX | Machine Learning

Senior-level Full Time

Mountain View, CA, USA

3d ago

Staff/Sr. ML Infrastructure Engineer, Foundation Model Compute Infra USD 175K-312K

C++ | Containers | Distributed Systems | Distributed Training | Distributed inference

Senior-level Full Time

Cupertino

3d ago

Senior Embedded AI Engineer (C/C++/Python) INR 2500K-4800K

C# | C++ | Compiler optimization | DSP | Debugging

Access to fitness clubs | Creche facility | Employee assistance program | Higher education opportunities | Life and accident insurance

Senior-level Full Time

IND - India Tech Center

3d ago

Applied Scientist, GenAI Catalog Intelligence, PRISM USD 171K-222K

Agentic Architectures | C++ | Confidence estimation | Deep learning | Experimental Design

Senior-level Full Time

Sunnyvale, California, USA

3d ago

Senior Research Engineer, Applied Robotics Navigation, DeepMind USD 207K-301K

Deep learning | Isaac Sim | Jetson | Language Models | Large Language Models

Senior-level Full Time

Mountain View, CA, USA

4d ago

Embedded Systems Engineer, Humanoid Robotics USD 144K-300K

ARM | BSP | Bare Metal | Board Support | Board Support Package

Senior-level Full Time

Boston, MA

4d ago

Intern - GenAI Benchmarking (MLE) / On-Device Model Deployment (SWE) KRW 45000K-55000K

AIMet | Automated testing | Cache Management | Code review | Context modeling

Entry-level Internship

Seoul, Korea, Republic of

4d ago

Principal Machine Learning Infrastructure Researcher USD 191K-250K

Deep learning | Ethernet | GPU Computing | Inference | Infiniband

Commuter benefits | Equity grants | Flexible hybrid workplace | Health care plan | Life insurance

Senior-level Full Time

Mountain View, CA

7d ago

Lead Cloud AI Engineer - Travel role USD 120K-150K

3PAO | AI APIs | Agent Builder | Air-gapped | Air-gapped environments

401k matching | Critical illness insurance | Dental insurance | Education assistance program | Employee assistance program

Senior-level Full Time

Washington, District of Columbia, United States

8d ago

Research Engineer, Pretraining, DeepMind USD 174K-253K

Artificial Intelligence | Fine Tuning | Inference Optimization | JAX | Language Models

Mid-level Full Time

New York, NY, USA

9d ago

Staff Software Engineer, Beyond Live, DeepMind USD 207K-301K

2D Games | 3D Games | Audio Tokenization | Audio video pacing | Audio/Video

Senior-level Full Time

New York, NY, USA; Mountain View, …

9d ago

DevOps/MLOps Specialist INR 3000K-4000K

AWS | Ansible | Apache Airflow | Apache Beam | Apache Kafka

Mid-level Full Time

bengaluru, India

9d ago

Technical Lead Manager, Machine Learning Runtime & Serving USD 251K-310K

Distributed Systems | Fault Tolerance | GPU Computing | JAX | Low Latency

Bonus program | Company benefits program | Equity incentive plan

Senior-level Full Time

Mountain View, California

9d ago

Head of AI & Agentic Platform Engineering USD 270K-500K

A B Rollback | A/B | A/B Testing | Access Control | Anomaly Detection

Hybrid work

Executive-level Full Time

USA - NY - Headquarters, United …

9d ago

AI Platform Engineer USD 136K-231K

Access Management | Bash | CI/CD | Container Security | Docker

401k | Dental insurance | Employee assistance program | Employee stock purchase program | Family care and bonding leave

Senior-level Full Time

USA-CA-Milpitas-KLA, United States

10d ago

Senior Software Engineer, ML Compiler, Frameworks and Performance INR 2500K-5000K

Accelerator hardware | C++ | Compiler Internals | Compiler optimization | Custom Kernels

Senior-level Full Time

Bengaluru, Karnataka, India

11d ago

Site Reliability Engineer, Inference Infrastructure USD 165K-267K

Amazon Web Services | C plus plus | Cloud platform | Distributed Systems | GPU

401k matching | Co working Benefit | Company offsite | Education and learning stipend | Health and dental benefits

Mid-level Full Time

Toronto

12d ago

Staff Software Engineer, Inference Infrastructure USD 160K-220K

Amazon Web Services | Azure | C++ | Cloud platform | Distributed Systems

401k | Annual company offsite | Arts and culture budget | Education and learning stipend | Health and dental benefits

Senior-level Full Time

San Francisco

12d ago

Software Engineer, On-Device Machine Learning USD 174K-253K

Android | Compiler development | Core ML | Data Processing | Debugging

Mid-level Full Time

Sunnyvale, CA, USA

14d ago

Senior Machine Learning Engineer (Computer Vision) USD 160K-214K

AWS | Azure | CI/CD | Computer Vision | Deep learning

Annual company retreat | Company-wide winter break | Education bonus | Equity participation | Optional in person events and meetups

Senior-level Full Time

Latin America R

16d ago

Staff Software Engineer, On-Device Machine Learning Infrastructure USD 207K-301K

APIs | Accelerated inference | Android | CPU | Data Processing

Senior-level Full Time

Sunnyvale, CA, USA

17d ago

Senior AI Infrastructure Engineer, LLM/AI Platforms USD 140K-215K

AWS | Airflow | Amazon SageMaker | Ansible | Batching

Competitive vacation and holidays | Comprehensive wellness programs | Employee networks | Paid adoption leave | Paid parental leave

Senior-level Full Time

USA VA Remote, United States R

17d ago

Senior Developer Relations Engineer, AI Infrastructure USD 163K-237K

C++ | Deep learning | Developer Content | GPU | Google Kubernetes

Senior-level Full Time

Sunnyvale, CA, USA; Cambridge, MA, USA

24d ago

Senior Solution Architect, AI Engineering, ASEAN Tech SGD 162K-225K

AWS Bedrock | AgentCore) | Agentic AI | Amazon SageMaker | Amazon Web Services

Senior-level Full Time

SGP

24d ago

Staff ML Engineer, Generative Model Performance & Efficiency USD 251K-310K

Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | Flax

Senior-level Full Time

Mountain View, California, United States, New …

24d ago

Machine Learning Systems Researcher USD 186K-240K

Deep learning | Ethernet | GPU | Inference | Infiniband

Commuter benefits | Equity grants | Flexible hybrid workplace | Health care plan | Life insurance

Mid-level Full Time

Mountain View, CA

24d ago

Global Head of Analytics and AI Platform Engineering, Managing Director CNY 300K-420K

Agentic AI | Analytics engineering | CUDA | Cloud Computing | Code review

Employee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer days

Executive-level Full Time

Hangzhou, China

28d ago

Global Head of Analytics and AI Platform Engineering, Managing Director CNY 300K-420K

Agentic AI | CUDA | Cloud Computing | Data Protection | Disaster Recovery

Employee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer days

Executive-level Full Time

Hangzhou, China

28d ago

Engineering Manager, ML Performance USD 207K-301K

Auto sharding | Benchmarking | CUDA | CUDA Performance | Compiler optimization

Senior-level Full Time

Sunnyvale, CA, USA; Kirkland, WA, USA

29d ago

Senior Delivery Consultant - Data & AI, AWS Professional Services GBP 80K-110K

AWS | Artificial Intelligence | Cloud Architecture | Data Engineering | Distributed Systems

Career growth | Flexible work arrangements | Mentorship | Training resources

Senior-level Full Time

London, England, GBR

1mo ago

Senior Computer Vision Engineer SGD 160K-214K

3D Reconstruction | 3D Vision | API Design | ARM | C plus plus

Senior-level Full Time

Singapore, Singapore, Singapore

1mo ago

Solution Architect - AI, Cross Industries INR 2000K-4800K

AgentCore) | Agentic Systems | Amazon Bedrock | Amazon Web Services | Cost Optimization

Senior-level Full Time

Mumbai, Maharashtra, IND

1mo ago

Software Engineer, Applied ML (Discovery, Recommendation & Search) USD 200K-300K

AWS | Automated testing | Azure | CI/CD | Data pipeline

Mid-level Full Time

Redwood City, CA

1mo ago

Sr. Delivery Consultant - AI/ML, AWS Professional Services USD 153K-239K

AWS | AWS AI | AWS AI Machine Learning | AWS AI Machine Learning Services | Analytics

Senior-level Full Time

Herndon, Virginia, USA

1mo ago

Research Engineer, Code RL (Reinforcement Learning) USD 500K-850K

Async Programming | CUDA | Code verification | Concurrent programming | Distributed Training

Flexible working hours | Generous vacation | Parental leave

Mid-level Full Time

San Francisco, CA | New York …

1mo ago

Applied Scientist II, Sponsored Products USD 172K-223K

AWS Neuron | Algorithms | C plus plus | Data Mining | Data Structures

401k matching | Dental insurance | EAP | Health insurance | Mental health support

Mid-level Full Time

New York, New York, USA

1mo ago

Machine Learning Engineer, Driver Understanding and Evaluation USD 170K-216K

C++ | Data Infrastructure | Embodied Agents | Flax | GPU

Mid-level Full Time

Mountain View, CA, USA

1mo ago

Senior Machine Learning Engineer USD 210K-361K

AWS | Aerospike | Apache Airflow | Apache Flink | Apache Spark

Dental benefits | Healthcare benefits | Hybrid work schedule | On-call rotation | Remote work on Fridays

Senior-level Full Time

Austin, Texas

1mo ago

Senior Staff AI Engineer USD 180K-240K

A3C | Actor-critic | Adaptive computation | Benchmarks | C plus plus

Senior-level Full Time

Los Altos, California,

1mo ago

Machine Learning Performance Engineer GBP 65K-80K

C# | C++ | CUDA | Compilers | GPU

Breakfast | Caregiver support | Dependent care | Fitness classes | Lunch provided

Mid-level Full Time

London, England, United Kingdom

1mo ago

Find jobs in AI/ML, Data Science and Big Data