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
209
Companies hiring for it
81
Remote share
10%
WHO'S HIRING

Companies with open roles mentioning JAX

Waymo34 open rolesMistral14 open rolesAnthropic10 open rolesCaylent10 open rolesLila Sciences9 open rolesWaabi6 open rolesTenstorrent5 open rolesSpaceX5 open rolesZoox4 open rolesIsomorphic Labs4 open rolesApptronik3 open rolesJumio3 open roles
APPEARS ALONGSIDE

Technologies mentioned in the same job posts

PyTorch×183Python×144TensorFlow×107C++×66Kubernetes×43CUDA×35Transformers×35Amazon Web Services×31Docker×28Triton×23
COMMON ROLES

Job titles that ask for JAX

  • Staff Machine Learning Engineer6
  • Ai/Ml Engineering Manager5
  • Principal Ai/Ml Architect5
  • Senior Machine Learning Engineer4
  • Machine Learning Engineer3
  • Member Of Technical Staff - Quantitative Research2
  • Machine Learning Engineer (Platform)2
  • Mid-Career Software 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.