1 to 25 of 29 Permanent JAX Jobs in the UK excluding London

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

ability to present complex ideas clearly Nice-to-have Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Previous research publications or submitted papers Why Join Us Gain hands-on experience in state-of-the-art AI research with real-world applications Collaborate with a team of world More ❯

ability to present complex ideas clearly Nice-to-have Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Previous research publications or submitted papers Why Join Us Gain hands-on experience in state-of-the-art AI research with real-world applications Collaborate with a team of world More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to publish More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to publish More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to publish More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this sounds More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this sounds More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this sounds More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for LLMs More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for LLMs More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for LLMs More ❯

facing developer workflows. Experience developing for and on Arm architectures. Experience profiling or debugging ML workloads. Experience designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT or HuggingFace Transformers. Experience collaborating closely with product or design teams. In Return: Our team offers an inclusive and highly collaborative environment, with work that spans a wide variety More ❯

reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty with cleaning More ❯

reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty with cleaning More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions • Fluency More ❯

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions • Fluency More ❯