joblaze

Sign in Create alert

← Back to jobs

TPU

8 jobs tagged TPU

SWE - Grids - Fixed Term Contract - 6 Months - London, UK

Google DeepMind

C++ Google Cloud Platform JAX TPU Python

London, UK

Published 2 weeks ago Apply →

TPU Kernel Engineer Remote Visa

low-precision inference Large Language Models collective communication algorithms ML systems TPU

San Francisco, USA

$280k–$850k/yr

Indexed 1 day ago Apply →

Research Engineer, Pretraining Scaling - London Visa

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

London, UK

£260k–£630k/yr

Indexed 1 day ago Apply →

Technical Program Manager, Infrastructure Remote Visa

AWS CI/CD Trainium AI tools TPU

San Francisco

$290k–$365k/yr

Indexed 1 day ago Apply →

Software Engineer, Inference Deployment Remote Visa

Trainium Rust TPU Python GPU

San Francisco

$320k–$485k/yr

Indexed 1 day ago Apply →

Research Engineer, Pretraining Scaling Visa

ML frameworks JAX model operations AI/ML large-scale distributed systems

San Francisco, USA

$350k–$850k/yr

Indexed 1 day ago Apply →

Engineering Manager, Inference Routing and Performance Remote Visa

continuous batching coordination planes prefill/decode disaggregation KV caching Trainium

San Francisco

$405k–$485k/yr

Indexed 1 day ago Apply →

LLM Inference Frameworks and Optimization Engineer

C++ TensorRT-LLM SGLang CUDA vLLM

San Francisco, Singapore, Amsterdam

$160k–$230k/yr

Indexed 1 day ago Apply →