"jax" Jobs

5 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: Python, Triton, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 5 of 5 results

Software Engineer - Triton

Join Graphcore's Triton team to enhance AI frameworks and optimize software for cutting-edge machine learning workloads.

Graphcore Gdańsk, Pomeranian Voivodeship, Poland Published 5 months ago

C++ Triton JAX PyTorch TensorFlow

Flexible on stack

Apply →

Software Engineer - Triton

Join Graphcore's Triton team to enhance AI frameworks and improve performance on cutting-edge hardware.

Graphcore Bristol, UK; Gdańsk, Pomeranian Voivodeship, Poland Published 2 weeks ago

C++ Triton JAX PyTorch TensorFlow

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research with a focus on hardware-aware algorithms and impactful implementations.

Graphcore Bristol, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on innovative AI hardware solutions.

Graphcore London, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on next-gen AI hardware.

Graphcore Cambridge, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →