26 JAX Development jobs

United States (Remote)

$184–356.5K / year

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

United States (Remote)

$224–431.3K / year

As a Global Account Manager, Developer Relations, you'll be responsible for the technical charter for advancing integration of NVIDIA's libraries,…

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.

Bristol, England, GB (On-site)

About the job Build the framework support that helps AI developers unlock Graphcore hardware.

United States (Remote)

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

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.

Bristol, England, GB (On-site)

About the job Build the framework support that helps AI developers unlock Graphcore hardware.

Shanghai, CN (On-site)

We're looking for outstanding AI systems software engineers to develop groundbreaking technologies across the inference systems software stack.

North America + 1 more (Remote)

Who are we? Cohere is the leading security-first enterprise AI company.

California, United States + 1 more (Remote)

$152–287.5K / year

Joining NVIDIA's DGX Cloud Team means contributing to the infrastructure that powers our innovative AI research.

Shanghai, CN (On-site)

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

Livingston, New Jersey, US (On-site)

$188–275K / year

CoreWeave is The Essential Cloud for AI™.

London, England, GB (Hybrid)

Who are we? Cohere is the leading security-first enterprise AI company.

Santa Clara, California, US (On-site)

$224–431.3K / year

NVIDIA’s DGX Cloud team builds the infrastructure behind large-scale AI training for NVIDIA’s own training workloads and for our customers.

Santa Clara, California, US (On-site)

$224–431.3K / year

NVIDIA’s DGX Cloud team builds the infrastructure behind large-scale AI training for NVIDIA’s own training workloads and for our customers.

NA, EMEA (Remote)

Who are we? Cohere is the leading security-first enterprise AI company.

Santa Clara, California, US (On-site)

$224–356.5K / year

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

Canada + 3 more (Remote)

Who are we? Cohere is the leading security-first enterprise AI company.

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.

Tokyo Prefecture, JP (Hybrid)

Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency.