184 JAX 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…

Germany (Remote)

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

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.

Zürich, CH (On-site)

CHF 221.3–507K / year

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years.

Belgrade, RS (On-site)

Perplexity is looking for a Search Machine Learning Engineer Intern to help build the next generation of advanced search technologies, with a focus…

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.

Shanghai, CN (On-site)

NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization.

Shanghai, CN (On-site)

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years.

Santa Clara, California, US (On-site)

$20.00 – $71.00 / hour

By submitting your resume, you acknowledge that your 2027 Deep Learning internship application will be processed in accordance with NVIDIA’s…

United States (Remote)

$152–287.5K / year

NVIDIA is a pioneer in accelerated computing, known for inventing the GPU and driving breakthroughs in gaming, computer graphics, high-performance…

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.

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.

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.

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.

Palo Alto, California, US (On-site)

$180–440K / year

SpaceXAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge.

Palo Alto, California, US (On-site)

$180–440K / year

SpaceXAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge.

United States (Remote)

$224–356.5K / year

NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High Performance Computing and Visualization.

Paris, Île-de-France, FR (On-site)

About Mistral Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute.