126 to 150 of 283 JAX Jobs in the UK

or Rust - Hands-on experience with parallel computing platforms such as CUDA, OpenMP, etc - Deep understanding of AI frameworks such as PyTorch, TensorFlow, and JAX, and their demands on underlying compute infrastructure, memory bandwidth, network interconnect, and storage as scale goes up - Knowledge of emerging AI hardware accelerators and architectures More ❯

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

industry, including knowledge of the landscape and Pure’s solutions with a focus on the following: Deep understanding of industry AI frameworks (pytorch, tensorflow, jax etc) Deep understanding on high performance, low latency, highly parallelized compute solutions at scale (compute, network, storage) Deep understanding of accelerated compute environments (GPU, ASIC More ❯

including designing new architectures, hands-on experimentation, analysis, and visualisation. Strong knowledge of linear algebra, calculus and statistics. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas. A passion for applying ML research to real world problems. Depending on your More ❯

years of Relevant Experience building and applying ML systems Proficiency in Python, Rust, C++, or Java Experience with ML frameworks like PyTorch, TensorFlow, or JAX Excellent communication skills—written, verbal, and visual AI Research Engineer - Well-Funded Start-Up #J-18808-Ljbffr More ❯

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

field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level hardware More ❯

field 5+ years of experience deploying AI/ML models in production settings Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization Experience with GPU architectures, cloud platforms, and hardware tradeoffs Familiarity with low-level hardware More ❯

years of relevant experience building and applying ML systems Proficiency in Python, Rust, C++, or Java Experience with ML frameworks like PyTorch, TensorFlow, or JAX Excellent communication skills—written, verbal, and visual Seniority level: Mid-Senior level Employment type: Full-time Job function: Research and Engineering Industries: Software Development #J More ❯

field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level hardware More ❯

field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level hardware More ❯

Proven experience scaling cloud-native systems (AWS, microservices, IaC) Kubernetes + Terraform expertise in production environments Solid understanding of ML deployment (e.g. PyTorch, TensorFlow, JAX) Frontend skills with React/TypeScript or similar Bonus: big data/real-time analytics/multi-cloud Why Join? Equity with serious upside and More ❯

Proven experience scaling cloud-native systems (AWS, microservices, IaC) Kubernetes + Terraform expertise in production environments Solid understanding of ML deployment (e.g. PyTorch, TensorFlow, JAX) Frontend skills with React/TypeScript or similar Bonus: big data/real-time analytics/multi-cloud Why Join? Equity with serious upside and More ❯

machine learning advancements. Requirements include strong programming skills in Python, C, C++, experience with deployment platforms, and familiarity with NLP, computer vision, TensorFlow, PyTorch, JAX, CUDA, LLMs, and related technologies. A degree in a relevant field and a solid AI R&D track record are essential. More ❯

with modern scientific software practices (version control, containerisation, continuous integration etc) and an enthusiasm for writing clear, performant code. Familiarity with ML frameworks (PyTorch, JAX or similar) and machine-learned potential energy surfaces. Previous computational drug discovery experience in industry. If you do not quite fulfil all of the requirements More ❯

APIs, FastAPIProven experience scaling cloud-native systems (AWS, microservices, IaC)Kubernetes + Terraform expertise in production environmentsSolid understanding of ML deployment (e.g. PyTorch, TensorFlow, JAX)Frontend skills with React/TypeScript or similarBonus: big data/real-time analytics/multi-cloudWhy Join? Equity with serious upside and rapidly rising More ❯

reinforcement learning, non-convex optimisation, approximate inference, NLP, or Bayesian methods Strong programming skills, ideally in Python, with experience using tools like NumPy, Pandas, JAX, PyTorch or TensorFlow A strong publication record in top-tier venues (e.g., NeurIPS, ICML, ICLR) or competitive performance in ML challenges such as Kaggle or More ❯

. Hands-on experience training and deploying learned policies for complex behaviors in robots or simulated characters. Fluency with modern ML toolchains (PyTorch/JAX, RL or behavior-cloning libraries, agentic frameworks, MCP, distributed training workflows). Solid software-engineering fundamentals - tested, documented, maintainable Python & C++. Demonstrated ability to debug More ❯

models over time in production considering model/data drifts In-depth experience with one or more deep neural network frameworks (e.g. PyTorch, Tensorflow, JAX) -A creative mindset, propensity to care deeply about the impact their team has and to encourage novel ways of critical thinking in their team Good More ❯

position such as post doctoral roles, a proven track record of publications, or contributions to machine learning codebases. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas Strong knowledge of linear algebra, calculus and statistics Experience working in a scientific More ❯

track record of launched products Fluency in Python and another language (e.g. C/C++) Hands on experience with relevant machine learning frameworks (Pytorch, Jax, Tensorflow) Skilled in software design, problem-solving, and debugging Experience optimizing models for resource-constrained devices is a plus What we offer Here are just More ❯

position such as post doctoral roles, a proven track record of publications, or contributions to machine learning codebases. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas Strong knowledge of linear algebra, calculus and statistics Experience working in a scientific More ❯

fit if you: Have 5+ years of industry-related experience Are proficient in Python and experienced with deep learning frameworks such as PyTorch or Jax Possess a strong software engineering background and enjoy working closely with researchers and engineers Enjoy pair programming Care about code quality, testing, and performance Are More ❯

LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech More ❯

LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech More ❯