Jax AI jobs

130 live jobs match "Jax AI jobs" right now. The median salary starts around $219,250 a year, and the most common tools are Jax, PyTorch, and TensorFlow.

AI Scientist - Biomedical Multimodal Modeling

Xaira TherapeuticsSouth San Francisco, California, United States$170k-$240k

pytorchjax

$170k-$240k4mo

Member of Technical Staff - Imagine Model

xAI Palo Alto, CA$180k-$440k

pytorchjax

$180k-$440k6mo

Member of Technical Staff, Integration/RL Team (Research Engineer)

CohereRemote · ParisCAD 250k-CAD 535k

pytorchjax

CAD 250k-CAD 535k12mo

ML Research Engineer, London

Isomorphic LabsLondon

pytorchjax

14mo

Member of Technical Staff, Post-Training

CohereRemote · LondonCAD 250k-CAD 535k

pytorchjax

CAD 250k-CAD 535k15mo

ML Runtime Optimization Engineer

Applied IntuitionSunnyvale$217k-$318k

pytorchjax

$217k-$318k18mo

Member of Technical Staff, Training Infra Engineer

CohereRemote · ParisCAD 250k-CAD 535k

pytorchjax

CAD 250k-CAD 535k18mo

Member of Technical Staff, Modeling

CohereRemote · LondonCAD 250k-CAD 535k

tensorflowjax

CAD 250k-CAD 535k22mo

Research Engineer, Data Foundations

Runway MLRemote · Remote$270k-$370k

pytorchjax

$270k-$370k23mo

Member of Technical Staff - RL Training Framework

xAI Palo Alto, CA$180k-$440k

jax

$180k-$440k1mo

Jax jobs by category