/

Point your AI agent at freehire and let it find you a job. A CLI and an MCP server over the whole job API — no browser.

Get the CLI →

What is JAX?

JAX is a machine learning library used for numerical computing and transforming mathematical functions. It enables automatic differentiation and compiles code to run efficiently on hardware accelerators like GPUs and TPUs.

237 open JAX jobs on freehire.

Named alongside JAX

Jobs asking for JAX

ASM

3 hours ago

Senior Engineer Product Intelligence (Materials Engineering – Chemistry & Thin Film Design)

NewBe an early applicant Senior

Step into a career with ASM, where cutting edge technology meets collaborative culture. For over 55 years ASM has been ahead of what’s next, at the forefront of innovation and what’s technologically possible. With…

AI API Cloud Data Pipelines Data Science +10 skills

Sign in to see your match

Apply
Atoms

13 hours ago

Senior Machine Learning Engineer

NewBe an early applicant North AmericaInternshipSenior
United States

Who we are Atoms is building the machines that power the next era of progress. Over the last decade, software has transformed the digital world. But the physical world, where food is made, minerals are mined, goods are…

AI Automation C++ CUDA Data Engineering +12 skills

Sign in to see your match

Apply
Atoms

13 hours ago

Staff Machine Learning Engineer

NewBe an early applicant North AmericaInternshipStaff
United States

Who we are Atoms is building the machines that power the next era of progress. Over the last decade, software has transformed the digital world. But the physical world, where food is made, minerals are mined, goods are…

AI Automation C++ CUDA Data Engineering +12 skills

Sign in to see your match

Apply
Anduril Industries

17 hours ago

Staff TLM, Machine Learning, Sentry Tower

NewBe an early applicant North AmericaFull-timeStaff
United States

Anduril Industries is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st…

AI CI/CD Computer Vision C++ Deep Learning +11 skills

Sign in to see your match

Apply
Focused

20 hours ago

Machine Learning Engineer (m/f/d)

NewBe an early applicant HybridEuropeFull-time
Germany

Focused Energy is building the future of clean energy through laser-driven inertial fusion. Backed by our $240M Series A — the largest fully secured Series A in the global fusion industry — we are scaling rapidly…

Anomaly Detection Ansys C++ Data Science Deep Learning +13 skills

Sign in to see your match

Apply
Focused

20 hours ago

Machine Learning Engineer

NewBe an early applicant HybridNorth AmericaFull-time
United States

Focused Energy is building the future of clean energy through laser-driven inertial fusion. Backed by our $240M Series A — the largest fully secured Series A in the global fusion industry — we are scaling rapidly…

Anomaly Detection Ansys C++ Data Science Deep Learning +13 skills

Sign in to see your match

Apply
Avride

22 hours ago

Machine Learning Engineer – Motion Planning & Prediction

NewBe an early applicant North America
United States

About the team We build the software that decides how our autonomous vehicles move through the world. Our systems predict the behavior of pedestrians, cyclists, and other vehicles, then plan trajectories that are safe,…

AI Analytics C++ Data Pipelines Deep Learning +7 skills

Sign in to see your match

Apply
Rhoda AI

4 days ago

Research Member of Technical Staff- Training Systems

NewBe an early applicant North AmericaFull-time
United States

At Rhoda AI, we’re building the next generation of generalist intelligent robots. We own the full robotics stack from high-performance hardware and robot systems to the infrastructure and state-of-the-art foundation…

AI CUDA Distributed Training JAX Machine Learning +3 skills

Sign in to see your match

Apply
Caylent

4 days ago

AI/ML Engineering Manager

NewBe an early applicant RemoteLATAMFull-timeSenior
Brazil

Caylent is an AI-first cloud services company that helps organizations turn ambitious ideas into meaningful business impact. As an AWS Premier Tier Services Partner and a charter member of Anthropic’s Claude Partner…

Agentic AI AI Airflow Analytics Anomaly Detection +66 skills

Sign in to see your match

Apply
Proxima Fusion

4 days ago

Software Engineer - Simulation Platform

NewBe an early applicant On-siteEuropeFull-time
Germany

WHO WE ARE Proxima Fusion is Europe’s fastest-growing fusion company and the continent’s best-funded fusion player, as well as the first spin-out from the Max Planck Institute for Plasma Physics (IPP). Backed by over…

Agentic AI AI Automation Cloud Data Pipelines +8 skills

Sign in to see your match

Apply

See all 237 →