Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

JAX Jobs with 401k in AI

Browse 4 JAX jobs with 401k in AI for Aug 2026 with roles open at xAI, Latitude AI, and Captivation paying $130K to $440K/year. Work as a ML Infrastructure Engineer, Senior Software Engineering Manager - Multi-Object Tracking & State Estimation, and Software Engineer 3 - Linux/Java/Jax-RS/REST/JSON/XML/Maven/Eclipse/Gitlab.

JAX is often requested for Research Engineer roles.

Receive emails of JAX Jobs with 401k in AI

AI Engineer Ansible Customer Success Manager Cypress Data Engineer Data Scientist Elasticsearch Engineering Manager Ethernet Full Stack Engineer HR JAX MongoDB NCCL OpenTelemetry Platform Engineer Product Manager Program Manager RDMA React Rust SEO Tableau TensorRT-LLM Triton TypeScript

Filter by benefits

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

Cities

San Francisco 12 Palo Alto 1 Cambridge 1 Glen Burnie 1 New York 1

Countries

United States 16 California 13 Massachusetts 1 Maryland 1 Canada 1 New York 1

Filter by type

Remote Full Time

C

Member of Technical Staff, Agentic Environments

Cohere / Canada ; New York, NY / Remote / Staff

Python PyTorch JAX

3D

LA

Senior Software Engineering Manager - Multi-Object Tracking & State Estimation

Latitude AI / Remote / Senior

Python PyTorch TensorFlow

$253K – $380K 6D

C

Software Engineer 3 - Linux/Java/Jax-RS/REST/JSON/XML/Maven/Eclipse/Gitlab

Captivation / Glen Burnie, United States

JAX Kubernetes Docker

$130K – $270K 6D

X

ML Infrastructure Engineer

xAI / Palo Alto, United States

Python PyTorch JAX

$180K – $440K 26D

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.)

What is the average salary for JAX jobs with 401k in AI?

4 active listings have salaries ranging from $130K to $440K/year.

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

Top companies hiring include xAI, Latitude AI, and Captivation.

Are there remote JAX jobs with 401k in AI?

Yes, 2 of 4 are remote.