TECHNOLOGY · AI & MACHINE LEARNING

JAX

Google's numerical computing library combining NumPy-like syntax with GPU/TPU acceleration and automatic differentiation for ML research.

Open roles mentioning it
288
Companies hiring for it
124
Remote share
8%
WHO'S HIRING

Companies with open roles mentioning JAX

Waymo31 open rolesCohere15 open rolesLila Sciences10 open rolesAnthropic10 open rolesWaabi6 open rolesJumio6 open rolesApptronik5 open rolesSandboxAQ5 open rolesCaylent5 open rolesAdaption5 open rolesIntrinsic5 open rolesRiot Games5 open roles
APPEARS ALONGSIDE

Technologies mentioned in the same job posts

PyTorch×267Python×209TensorFlow×164C++×85CUDA×58Amazon Web Services×48Docker/Kubernetes×47Transformers×38Triton×33TensorRT×32
COMMON ROLES

Job titles that ask for JAX

  • Machine Learning Engineer7
  • Ai/Ml Engineering Manager5
  • Staff Machine Learning Engineer4
  • Senior Machine Learning Engineer4
  • Senior Engineering Manager - Machine Learning4
  • Ai Modeling Engineer3
  • Internship - Search Machine Learning Engineer2
  • Genesis-World: Core Simulation Engine Engineer2
GO DEEPER

Search these companies

Every technology on this page is extracted from live job descriptions — not self-reported marketing. Filter companies by tech stack or create an account to track the ones hiring for yours.