Full Time JAX AI Jobs in United States (2 New)

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

Cities

San Francisco 4 Seattle 3 Cambridge 2 Cambridge 1 San Jose 1 Washington 1

Countries

United States California 6 Washington 3 United Kingdom 2 Massachusetts 1 District of Columbia 1

Filter by type

Remote Full Time Internship

Senior Staff Software Engineer, Storage

Crusoe / Sunnyvale, United States / Staff

PyTorch TensorFlow JAX

$245K – $290K 1D

Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)

Cohere / New York, NY ; Toronto, CA / Remote / Staff

PyTorch JAX RLHF

Research Engineer, Safety & Alignment

Cognition / San Francisco, United States

Python PyTorch JAX

TML

RL Systems - Infra

Thinking Machines Lab / San Francisco, United States

PyTorch JAX RLHF

$425K – $475K 9D

Software Engineer, Numerical Computing

CHAOS Industries / Washington, United States

Python PyTorch JAX

$150K – $210K 9D

SSI

Staff Engineer, ML Profiling Tools

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

TensorFlow JAX Excel

$163K – $253K 14D

TAI

Research Engineer, Climate Modeling

The Allen Institute for AI / Seattle, United States

Python JAX

$147K – $220K 21D

TAI

Research Scientist, Climate Modeling

The Allen Institute for AI / Seattle, United States

Python JAX

$167K – $261K 21D

TAI

Young Investigator, Climate Modeling

The Allen Institute for AI / Seattle, United States

JAX

$160K+ 21D

FPI

Senior Director / Vice President, Machine Learning (Generative Chemistry)

Flagship Pioneering, Inc. / Cambridge, United States / Senior

PyTorch TensorFlow JAX

$279K – $413K 21D

TML

Research, Mid Training

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch TensorFlow JAX

$350K – $475K 25D

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 the average salary for Full Time JAX jobs?

12 active listings have salaries ranging from $146K to $475K/year.

Which companies are hiring for Full Time JAX jobs?

Top companies hiring include Cohere, CHAOS Industries, and Cognition.

Are there remote Full Time JAX jobs?

Yes, 2 of 12 are remote.