1 to 25 of 34 Permanent JAX Jobs in London

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

customer-facing communication across scientific and engineering audiences. Experience in pharma, biotech, medtech, or healthcare environments (ideal). MSc in Computer Science or equivalent experience. Bonus Familiarity with PyTorch, JAX, Hugging Face, and AWS ML services. Experience optimizing models for low-latency inference. Open-source contributions and/or startup experience. Why This Is Unique Globally competitive salary + equity. More ❯

customer-facing communication across scientific and engineering audiences. Experience in pharma, biotech, medtech, or healthcare environments (ideal). MSc in Computer Science or equivalent experience. Bonus Familiarity with PyTorch, JAX, Hugging Face, and AWS ML services. Experience optimizing models for low-latency inference. Open-source contributions and/or startup experience. Why This Is Unique Globally competitive salary + equity. More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity More ❯

of high-quality research, evidenced by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong analytical and critical thinking skills 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 ❯

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 ❯

Collaborate cross-functionally with hardware, software and research teams. Requirements Strong background in machine learning algorithm development and optimisation. Proficiency in Python and modern ML frameworks such as PyTorch, JAX or TensorFlow. Experience with data modelling, preprocessing and performance benchmarking. Understanding of ML evaluation metrics and model analysis methods. Excellent communication skills and collaborative mindset. 3+ years’ experience in ML More ❯

Collaborate cross-functionally with hardware, software and research teams. Requirements Strong background in machine learning algorithm development and optimisation. Proficiency in Python and modern ML frameworks such as PyTorch, JAX or TensorFlow. Experience with data modelling, preprocessing and performance benchmarking. Understanding of ML evaluation metrics and model analysis methods. Excellent communication skills and collaborative mindset. 3+ years’ experience in ML More ❯

data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯

data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯

team at a company 8 months ahead of roadmap What they’re looking for: 3–6 years’ experience writing production-grade code Strong foundation in machine learning (PyTorch/JAX/TensorFlow) Deep understanding of LLMs and RAG, not just LangChain wrappers Bonus: Research internships, papers, or early-stage AI product builds Someone who wants to build real things, not More ❯

team at a company 8 months ahead of roadmap What they’re looking for: 3–6 years’ experience writing production-grade code Strong foundation in machine learning (PyTorch/JAX/TensorFlow) Deep understanding of LLMs and RAG, not just LangChain wrappers Bonus: Research internships, papers, or early-stage AI product builds Someone who wants to build real things, not 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 ❯

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 ❯

Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯

Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯