"jax" Jobs

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

Showing 10 of 10 results

Perplexity AI

Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.

Perplexity AI London Published 5 months ago
Flexible on stack
DeepL

Join DeepL as a Senior Research Scientist to innovate in reinforcement learning and shape the future of AI technology.

DeepL London Published 1 month ago
Flexible on stack
Perplexity AI

Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.

Perplexity AI London Published 5 months ago
Flexible on stack
Coreweave

Join CoreWeave as an Account Solution Architect to support AI labs and enterprises with GPU infrastructure and MLOps solutions across Northern EMEA.

Coreweave London, UK Published 3 months ago
Flexible on stack
Graphcore

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
Flexible on stack