Software Engineering jobs using JAX
178 live jobs match "Software Engineering jobs using JAX" right now. 159 use, work on, or build AI, not just mention it in the posting, and the median salary starts around $170,000 a year, and the most common tools are JAX, PyTorch, and TensorFlow.
New software engineering jobs using jax by email
$184k-$288kUS, CA, Santa Clara3w
pytorchtensorflowjax+4
India, Bengaluru, Karnataka3w
openaianthropicpytorch+5
Sunnyvale, CARemote3w
openaipytorchjax+2
Sunnyvale, CARemote3w
openaipytorchjax+3
$148kUS, WA, Seattle3w
llamadeepseekpytorch+5
$127k-$237kUnited States of America, Eagan, Minnesota3w
openaianthropicpytorch+7
$85/hrMountain View, CA, USA4w
tensorflowjaxpython+1
$129kUS, CA, Cupertino4w
pytorchtensorflowjax+1
$272k-$340kSan Francisco Bay Area or Remote (U.S.)Remote4w
pytorchtensorflowjax+4
$165k-$224kUS, CA, Cupertino4w
pytorchjaxaws-trainium+1
$193k-$262kUS, CA, Cupertino4w
llamadeepseekvllm+4
$193k-$262kUS, CA, Cupertino4w
pytorchjaxaws-trainium+1
$193k-$262kUS, CA, Cupertino4w
pytorchtensorflowjax
China, Shanghai4w
vllmpytorchjax+6
$144k-$194kUS, WA, Seattle1mo
pytorchtensorflowjax
$184k-$288kUS, CA, Santa Clara1mo
vllmpytorchjax+6
$224k-$357kUS, CA, Santa Clara1mo
pytorchjaxpython
$152k-$242kUS, CA, Santa Clara1mo
openaipytorchtensorflow+7
CAD 101kCA, ON, Toronto1mo
pytorchjaxaws-trainium
$320k-$485kSan Francisco, CA1mo
bedrockpytorchtensorflow+2
$350k-$475kSan FranciscoRemote1mo
pytorchtensorflowjax+1
$110k-$204kUnited States of America, Eagan, Minnesota1mo
openaianthropicpytorch+7
CAD 285k-CAD 340kCanadaRemote1mo
pytorchtensorflowjax+5
$165k-$224kUS, CA, Cupertino1mo
llamadeepseekpytorch+9
$168k-$227kUS, WA, Seattle1mo
pytorchtensorflowjax+2
San Francisco1mo
pytorchjaxkubernetes+1
Switzerland, Zurich1mo
pytorchtensorflowjax+3
$184k-$288kUS, CA, Santa Clara1mo
vllmpytorchjax+3
China, Shanghai4w
pytorchtensorflowjax+4
Get new software engineering jobs using jax by email
13 new software engineering jobs using jax in the last 7 days, as of Oct 24, 2026. One email a week, only when a new one matches this page.
Confirm by email first. Unsubscribe in one click.
Sources: O*NET OnLine (U.S. Dept. of Labor): Software Developers (SOC 15-1252.00), The JAX Authors: JAX documentation.