"tpu" Jobs

6 open tech roles matching “tpu”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: PyTorch, TPU, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 6 of 6 results

TPU Kernel Engineer Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 1 year ago

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

Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 10 months ago

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Apply →

Research Engineer, Pretraining Scaling Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 10 months ago

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

Apply →

Technical Program Manager, Hardware Systems Remote Visa

Own the execution of hardware systems from concept through production deployment in a fast-paced AI environment.

Anthropic San Francisco, CA | New York City, NY $365k–$435k/yr Published 2 weeks ago

hardware systems supply chain Firmware ODM CM

Apply →

LLM Inference Frameworks and Optimization Engineer

Together AI San Francisco, Singapore, Amsterdam $160k–$230k/yr Published 1 year ago

C++ TensorRT-LLM SGLang CUDA vLLM

Apply →

Research Engineer, Interpretability Remote Visa

Anthropic San Francisco, CA $315k–$560k/yr Published 9 months ago

JAX CUDA Interpretability Java Go

Apply →