jax

Recent positions

• Sr. Software Engineer (Vehicle Engineering)

  spacex · 6d ago

  $160k – $225k

  senior · on-site · full time · Hawthorne

  python pytorch docker kubernetes postgresql tensorflow +1
  Apply

• AI engineer

  writer · 6d ago

  $152k – $315k

  senior · hybrid · full time · San Francisco

  python pytorch aws gcp tensorflow azure +1
  Apply

• AI engineer (UK)

  writer · 6d ago

  senior · hybrid · full time · London

  python pytorch aws gcp tensorflow azure +1
  Apply

• Software Engineer (Large Scale Training)

  lightricks · 6d ago

  mid · on-site · full time · Jerusalem

  c++ python cuda kubernetes jax metal +2
  Apply

• Lead AI Research Scientist

  workato · 9d ago

  $357k

  director · on-site · full time · San Francisco

  pytorch cuda jax transformer unstructured structured
  Apply

• Junior Quantitative Researcher (Fresh STEM PhD graduates are welcome)

  binance · 10d ago

  junior · remote · full time · Hong Kong

  python pytorch jax pandas numpy
  Apply

• Research Engineer, Asta

  thealleninstitute · 11d ago

  $118k – $178k

  mid · on-site · full time · Seattle

  python pytorch aws docker gcp tensorflow +3
  Apply

• Machine Learning Intern/Co-op (Fall, 2026)

  cohere · 11d ago

  intern · remote · internship

  python cuda tensorflow jax tpu
  Apply

• Member of Technical Staff, Distributed Systems

  psi · 12d ago

  senior · on-site · full time · Boston

  python pytorch aws terraform docker kubernetes +14
  Apply

• Research Scientist (Singapore)

  cantina · 12d ago

  senior · on-site · full time · Singapore

  pytorch opencv docker kubernetes jax airflow +4
  Apply

• ML Infrastructure Engineer

  nebius · 12d ago

  senior · on-site · full time · Amsterdam, Berlin, London

  pytorch cuda docker kubernetes jax
  Apply

• Staff Machine Learning Engineer

  motional · 13d ago

  $205k – $272k

  staff · hybrid · full time · Boston, Las Vegas, Pittsburgh

  python pytorch aws gcp tensorflow jax
  Apply

• Staff Machine Learning Engineer

  motional · 13d ago

  $205k – $272k

  staff · hybrid · full time · Boston, Las Vegas, Pittsburgh

  python pytorch aws gcp tensorflow azure +1
  Apply

• Staff Machine Learning Engineer

  motional · 13d ago

  $205k – $272k

  staff · hybrid · full time · Boston, Las Vegas, Pittsburgh

  python pytorch aws gcp tensorflow azure +1
  Apply

• Staff Machine Learning Engineer

  motional · 13d ago

  $205k – $272k

  staff · hybrid · full time · Boston, Las Vegas, Pittsburgh

  python pytorch aws gcp tensorflow azure +1
  Apply

• AI/ML Engineering Manager

  caylent · 13d ago

  $152k – $234k

  director · remote · full time

  pytorch aws docker kubernetes tensorflow azure +8
  Apply

• Software Engineer, Perception

  intrinsicrobotics · 13d ago

  mid · on-site · full time · Singapore

  python pytorch cuda tensorflow jax
  Apply

• Senior Machine Learning Scientist - Cellular Modelling

  relationrx · 16d ago

  senior · on-site · full time · London

  python pytorch jax
  Apply

• Machine Learning Engineer, Speech LLM Training - San Francisco

  plaud · 16d ago

  $200k – $540k

  senior · hybrid · full time · San Francisco

  pytorch jax
  Apply

• Computer Vision Researcher (VLM)

  niantic-spatial · 17d ago

  £90k – £120k

  senior · hybrid · full time · London

  pytorch ros jax
  Apply