14 JAX Developer jobs

Santa Clara, California, US (On-site)

$272–431.3K / year

JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast.

Switzerland + 4 more (Remote)

PLN 375–650K / year

NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using…

Switzerland + 4 more (Remote)

PLN 292.5–650K / year

NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using…

Gdańsk, Pomeranian Voivodeship, PL (On-site)

PLN 260.4–352.2K / year

Salary Range: PLN Range: 260,400 - 352,200 About the job Build the framework support that helps AI developers unlock Graphcore hardware.

Bengaluru, Karnataka, IN (On-site)

NVIDIA is the leading full-stack accelerated computing company, powering the next wave of generative AI, agentic AI, deep learning, data science,…

United States (Remote)

At Hugging Face, we're on a journey to democratize good AI.

Santa Clara, California, US (On-site)

$108–195.5K / year

Today, we’re tapping into the unlimited potential of AI to define the next era of computing.

United States (Remote)

$152–287.5K / year

We are seeking a mission-driven Developer Relations Manager focused on Foundational AI Research to engage leading academic labs advancing the next…

Germany + 5 more (Remote)

PLN 375–650K / year

NVIDIA’s accelerated computing platform is the foundation of modern HPC and AI.At the core of this platform are the CUDA Core Libraries.

Santa Clara, California, US (On-site)

$184–356.5K / year

Become a part of NVIDIA, the industry front runner in AI computing, and help us revolutionize AI-powered solutions.

Livingston, New Jersey, US (On-site)

$188–275K / year

CoreWeave is The Essential Cloud for AI™.

San Francisco, California, US (On-site)

$350–475K / year

About Thinking Machines The mission of Thinking Machines is to build AI that extends human will and judgment.

France (Remote)

NVIDIA is seeking a highly influential Generative AI Technical Engagement Lead to evangelize for, drive, and support the seamless adoption and…

United States (Remote)

$184–356.5K / year

We are looking for an experienced and highly motivated software professional to work on pioneering initiatives and projects at the intersection of…