6 of 6 Remote JAX Jobs in the UK excluding London

frameworks (e.g., PyTorch, TensorFlow)Solid grasp of ML techniques; independent and team-oriented mindset Preferred : Experience in LLMs, reinforcement learning, or chip designFamiliarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applicationsCollaborate with top researchers in a global teamFlexible hybrid work environmentOpportunities More ❯

from a top-tier university. • Solid experience in training neural networks, tuning hyperparameters, and working with frameworks like PyTorch. • Knowledge of differentiable programming (e.g., JAX) or geometric deep learning (e.g., PyTorch Geometric) is a plus. • Strong foundation in mathematical concepts such as stochastic calculus, optimisation, and simulation techniques. • Proficiency in More ❯

focus on GenAI Strong expertise in Stable Diffusion, Flux, and similar architectures Experience fine-tuning and optimising visual models Hands-on experience with PyTorch, JAX, or TensorFlow Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT) Experience deploying ML models using Kubernetes and/or Docker Strong experience with Python and More ❯

focus on GenAI 🚀 Strong expertise in Stable Diffusion, Flux, and similar architectures 🚀 Experience fine-tuning and optimising visual models 🚀 Hands-on experience with PyTorch, JAX, or TensorFlow 🚀 Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT) 🚀 Experience deploying ML models using Kubernetes and/or Docker 🚀 Strong experience with Python and More ❯

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯

Machine Learning Scientists/Engineers, Cambridge, UK Astex Pharmaceuticals is a global leader in innovative drug discovery, utilizing its proprietary Fragment-Based Drug Discovery platform to develop new drug candidates, with several progressing in clinical trials. The company focuses on More ❯