15 of 15 Permanent Google JAX Jobs in the UK excluding London

Python, R, C, C++, or Java, and their associated security considerations. • Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. • Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. • Strong understanding of More ❯

inference, causal reasoning & inference, transfer & multi-task learning, diffusion models, graph neural networks, active learning; Experience writing production-quality code with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, or similar; Experience with multi-GPU and distributed training at scale; A team player who thrives in collaborative environments and is committed to enabling colleagues to reach their full … physics, or computational biology, or equivalent experience; Experience in developing machine learning models; Very strong programming skills, including experience with Python and deep learning libraries such as PyTorch or JAX; Experience in large-scale distributed optimization of machine learning models across multiple GPUs and nodes. Preferred Qualifications - Familiarity with biological data formats, concepts, and computational models; Experience in cell More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

About Us Deep Genomics is at the forefront of using artificial intelligence to transform drug discovery. Our proprietary AI platform decodes the complexity of genome biology to identify novel drug targets, mechanisms, and genetic medicines inaccessible through traditional methods. With More ❯

Permanent Role Location : Cambridge Hybrid Policy: 3 days onsite and 2 days working from home weekly Role and Responsibilities As anAI Research Scientist, you will be part of a dynamic team working on next-generation machine learning technologies for real More ❯

Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of More ❯

that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C More ❯

that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C More ❯

that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C More ❯

/EE or related research experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU) Experience with distributed and More ❯

or a related field, plus 3+ years of relevant experience. Demonstrated expertise applying deep learning to omics data at scale. Proficiency in Python and modern deep learning frameworks (PyTorch, JAX, or similar). Experience designing and operating data pipelines for genomics, transcriptomics, proteomics, or metabolomics. Solid grasp of causal inference, experimental design, and statistics. Familiarity with software engineering best More ❯

supervised pre training, Vision Transformers, or vector quantization methods. Hands on work with large medical or scientific imaging datasets and establishing evaluation tasks and metrics. Proficiency in PyTorch or JAX and experience running distributed training on GPUs or cloud HPC resources. Strong software engineering habits: version control, testing, continuous integration, code review, and containerization. Clear communication and ability to More ❯

About the role: Join a mission-focused technology consultancy delivering breakthrough AI and machine learning solutions to complex, real-world problems. You will apply your curiosity, creativity, and rigour to design and deploy ML systems across diverse projects, working in More ❯

modern AI/ML methods (e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable AI-driven workflows and tools. More ❯