Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

TPU Jobs with 401k in AI

Browse 3 TPU jobs with 401k in AI for Sep 2026 with roles open at Samsung Semiconductor, Inc., Cohere, and Baseten. Work as a Senior Staff Engineer - AI Workloads & Storage, Senior Software Engineer, GPU Infrastructure (HPC), and Global Capacity Manager - TPU Focus.

TPU is often requested for Software Engineer jobs roles.

Receive emails of TPU Jobs with 401k in AI

Cypress Data Engineer Data Scientist Ethernet Head of Design Member of Technical Staff MongoDB NCCL Network Engineer Product Designer Product Lead PyTorch RDMA React Rust Site Reliability Engineer Sketch Solutions Engineer Support Engineer TPU Tableau Talent Acquisition Specialist TensorRT-LLM Transformers Triton TypeScript

Filter by benefits

🚑 Medical Insurance 🦷 Dental Insurance 🤓 Vision Insurance 💰 401k 💰 Equity 🏖 Unlimited Vacation 🏖 Paid Time Off ⏰ Async 🏔 Company Retreats 🖥 Home Office Budget 📚 Learning Budget 💪 Free Gym Membership 🧘 Mental Wellness Budget ✈️ Visa Sponsorship 📦 Relocation 💊 FSA 🏥 HSA

Cities

San Francisco 6 San Jose 4 New York 3 Ho Chi Minh City 1 London 1

Countries

United States 12 California 10 New York 3 Vietnam 1 United Kingdom 1 Canada 1

Filter by type

Remote Full Time

C

Senior Software Engineer, GPU Infrastructure (HPC)

Cohere / United States ; Canada / Hybrid / Senior

Python PyTorch TensorFlow

13D

SSI

Senior Staff Engineer - AI Workloads & Storage

Samsung Semiconductor, Inc. / San Jose, United States / Staff

vLLM TensorRT C++

$189K – $301K 20D

B

Global Capacity Manager - TPU Focus

Baseten / San Francisco, United States / Hybrid

Python JAX Kubernetes

6D

Frequently asked questions

What happens to your 401k when you leave a job?

You have four options: leave the funds in your former employer's plan if the balance qualifies, roll it over into your new employer's 401k, roll it over into an IRA, or cash it out (which triggers income tax and, if you're under 59½, a 10% early withdrawal penalty). If your balance is under $7,000, your former employer can automatically roll it into an IRA on your behalf without your consent. (Figures current as of July 2026.)

Which companies are hiring for TPU jobs with 401k in AI?

Top companies hiring include Samsung Semiconductor, Inc., Cohere, and Baseten.