176 to 200 of 240 JAX Jobs in the UK

or multimodal architectures. Experience applying ML to large multimodal biological datasets (genomics, transcriptomics, epigenomics, proteomics, single-cell data). Exceptional coding skills (Python, PyTorch, JAX, TensorFlow) and familiarity with scalable training frameworks. Experience managing large-scale distributed training across GPUs/TPUs. Background in generative AI, large language models, mechanistic More ❯

techniques, model architecture, optimizer, learning rate schedules, etc. Thorough understanding of Python tools and libraries, keen to offer advice on best practices Experience using Jax, Tensorflow or similar ML frameworks a huge plus Benefits: Market-leading salaries Generous benefits package, including physical & mental health benefits, excellent holiday entitlement, significant parental More ❯

techniques, model architecture, optimizer, learning rate schedules, etc. Thorough understanding of Python tools and libraries, keen to offer advice on best practices Experience using Jax, Tensorflow or similar ML frameworks a huge plus Benefits: Market-leading salaries Generous benefits package, including physical & mental health benefits, excellent holiday entitlement, significant parental More ❯

of the latest trends in AI/ML research and development Good familiarity with one of the popular ML frameworks such as PyTorch, TensorFlow, JAX, ONNX, or PaddlePaddle A history of developer engagement or participation in an active open-source development community Experience in Product Owner roles, or a desire 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 ❯

driving stack. Experience MS or higher degree in a related field, or equivalent industry experience. Professional experience with ML frameworks such as PyTorch (preferred), Jax, or TensorFlow. Knowledge of debugging and profiling neural networks on NVIDIA CUDA. Experience with architectures for object detection and 3D perception. Experience in ML workflows More ❯

to-end stack. Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Knowledge of debugging and profiling deep neural networks on NVIDIA CUDA stack Experience in state of the art architectures for object More ❯

You Proficient hands-on experience training deep learning models, ideally at scale (multi-GPU, multi-node) Strong engineering skills in Python and PyTorch (or JAX); comfortable with distributed training setups Familiarity with dataset curation, versioning, and large-scale data management Hands-on experience running and analysing hyperparameter sweeps Deep curiosity More ❯

Ops lifecycle, tools, and frameworks Nice to have: Experience supporting machine learning research activities Familiarity with python-based ML frameworks like TensorFlow, PyTorch, or JAX Experience with Google Cloud Platform (GCP) Culture and values What does it take to be successful at IsoLabs? It's not about finding people who More ❯

to-end stack. Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Knowledge of debugging and profiling deep neural networks on NVIDIA CUDA stack Experience in state of the art architectures for object More ❯

transformers, generative adversarial networks, and diffusion models. You are proficient with: Deep learning frameworks (Pytorch gives bonus points, alternatively other frameworks like Tensorflow or JAX), Python 3, Numpy, OpenCV, Docker & Kubernetes. You are able to pick up knowledge of the state-of-the-art of a specific ML topic quickly More ❯

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 More ❯

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 More ❯

and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations, with More ❯

Organisation/Company KINGS COLLEGE LONDON Research Field Computer science Mathematics Physics Researcher Profile First Stage Researcher (R1) Country United Kingdom Application Deadline 1 Jun 2025 - 00:00 (UTC) Type of Contract Other Job Status Full-time Is the job More ❯

the efficiency of machine learning models through techniques like quantization, pruning, and other optimization methods. Harness the power of GPUs through CUDA, Python, or Jax to accelerate simulations and training processes. Qualifications: Solid understanding of control theory, with a focus on manipulation and locomotion tasks. Proficiency in Python/C++ More ❯

the efficiency of machine learning models through techniques like quantization, pruning, and other optimization methods. Harness the power of GPUs through CUDA, Python, or Jax to accelerate simulations and training processes. Qualifications: Solid understanding of control theory, with a focus on manipulation and locomotion tasks. Proficiency in Python/C++ More ❯

in Python software development, demonstrated by published software projects (e.g., GitHub). Experience in developing and implementing deep learning systems (e.g., in PyTorch or JAX). Preferred: Experience working with graph data, reinforcement learning, and/or generative models. Experience with molecular data in real-world drug discovery. Track record More ❯

the efficiency of machine learning models through techniques like quantization, pruning, and other optimization methods. Harness the power of GPUs through CUDA, Python, or Jax to accelerate simulations and training processes. Qualifications: Solid understanding of control theory, with a focus on manipulation and locomotion tasks. Proficiency in Python/C++ More ❯

robot behavior. Enhance the efficiency of machine learning models through techniques like quantization, pruning, and other optimization methods. Utilize GPUs with CUDA, Python, or Jax to accelerate simulations and training processes. Qualifications: Solid understanding of control theory, especially manipulation and locomotion tasks. Proficiency in Python/C++ and deep learning 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 ❯

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 ❯

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 ❯

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 ❯

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 ❯