Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

JAX Jobs with Equity in AI

Browse 6 JAX jobs with Equity in AI for Oct 2026 with roles open at Discord, Anthropic, and CHAOS Industries paying $150K to $850K/year. Work as a Staff Software Engineer, Machine Learning (Consumer Revenue), Research Engineer / Research Scientist, RL Frontiers, and Software Engineer, Numerical Computing.

JAX is often requested for Software Engineer jobs roles.

Receive emails of JAX Jobs with Equity 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 Product Manager Program Manager RDMA React Rust SEO Sketch 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 💊 FSA 🏥 HSA

Cities

San Francisco 11 Seattle 3 Santa Clara 2 Amsterdam 2 Toronto 2 Cambridge 2

Countries

United States 21 California 15 Washington 3 Netherlands 2 Canada 2 United Kingdom 2

Filter by type

Remote Full Time Internship

C

Senior Staff Software Engineer, Storage

Crusoe / Sunnyvale, United States / Staff

PyTorch TensorFlow JAX

$245K – $290K 1D

G

Senior Machine Learning Engineer (Large Systems)

Graphcore / Bristol, GB ; Cambridge, GB / Senior

Python PyTorch JAX

7D

CI

Software Engineer, Numerical Computing

CHAOS Industries / Washington, United States

Python PyTorch JAX

$150K – $210K 9D

PLI

Senior Software Engineer

Planet Labs Inc. / Ljubljana, Slovenia / Senior

€58K – €72K 9D

A

Research Engineer / Research Scientist, RL Frontiers

Anthropic

Python PyTorch JAX

$500K – $850K 17D

D

Staff Software Engineer, Machine Learning (Consumer Revenue)

Discord / San Francisco, United States / Remote / Staff

Python PyTorch TensorFlow

$272K – $340K 25D

Frequently asked questions

What is an equity grant in a job offer?

An equity grant gives you partial ownership in the company, usually as stock options (the right to buy shares at a fixed price after vesting) or RSUs (shares awarded outright once vesting conditions are met). Vesting is typically time-based over 3-4 years, often with a 1-year cliff before anything vests.

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

6 active listings have salaries ranging from $150K to $850K/year.

Which companies are hiring for JAX jobs with Equity in AI?

Top companies hiring include Discord, Anthropic, and CHAOS Industries.

Are there remote JAX jobs with Equity in AI?

Yes, 1 of 6 are remote.