76 to 100 of 142 JAX Jobs in London

custom in-house). Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

custom in-house). Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

custom in-house). Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

custom in-house). Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

custom in-house). Strong grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

custom in-house). Strong grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

custom in-house). Strong grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

be an advantage: Hands-on experience with prompt engineering and post-training techniques for large language models, including data pipeline and evaluation methodologies. Experience with ML frameworks such as JAX, PyTorch, or TensorFlow. Experience training and deploying large-scale models At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities to create extraordinary impact. More ❯

or more of: deep learning, 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 similar platforms The ability to More ❯

languages such as Python, C++ 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 - Experience with containerization and orchestration More ❯

Robotics or related Field 4+ 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 ❯

Science, Engineering, or a related 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 tuning and custom CUDA kernel More ❯

Robotics or related field 4+ 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-18808-Ljbffr More ❯

Science, Engineering, or a related 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 tuning and custom CUDA kernel More ❯

existing products with the latest 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 ❯

learning using deep learning techniques, 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 experience: project supervision, leadership, or More ❯

or more of: deep learning, 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 similar platforms The ability to More ❯

or more of: deep learning, 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 similar platforms The ability to More ❯

and requirements for an end-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) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for end-to-end autonomous driving Experience More ❯

similarly strong academic background in machine learning, with 2+ years of hands-on experience in LLM fine-tuning, RAG systems, and prompt engineering. Proficiency with frameworks like PyTorch, Tensorflow, JAX, Pandas, and OpenCV. A solid understanding of transformer models and their common variants, loss functions, and pitfalls. A quick learner with a structured, organized approach to problem-solving. Excellent communication More ❯

and ability to quickly pick up new technologies. Exposure to Git, Unix, SQL, or any high-performance computing language is a plus but not required. Exposure to PyTorch/Jax is a plus but not required Our benefits To help you stay energized, engaged and inspired, we offer a wide range of employee benefits including: retirement investment and tools designed More ❯

strong academic background in machine learning, with 2+ years of hands-on experience in with LLM fine-tuning, RAG systems, and prompt engineering. Proficiency with frameworks like PyTorch, Tensorflow, JAX, Pandas, and OpenCV. A solid understanding of transformer models and their common variants, loss functions, and pitfalls. A quick learner with a structured, organized approach to problem-solving. Excellent communication More ❯

learning using deep learning techniques, 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 experience: project supervision, leadership, or More ❯

Research Engineer, Machine Learning (Horizons) London, UK About Anthropic Anthropic’s mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team More ❯

At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯