13 of 13 JAX Jobs in the UK excluding London

language processing, including hands-on experience with transformer architectures (e.g., GPT, BERT, T5). Expertise in deep learning frameworks such as TensorFlow , PyTorch , or JAX . Strong knowledge of machine learning algorithms and their application to NLP tasks. Experience with cloud platforms such as AWS, GCP, or Azure for model ...

learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

theory, statistics, optimisation, etc) Proficiency in Python (required) and C++ (preferred for robotics integration) Experience with ML frameworks such as PyTorch (preferred), TensorFlow, or JAX Proven track record building production ML systems, not just research prototypes Strong software engineering fundamentals: modular design, testing, and version control Experience with cloud/ ...

probability, optimisation, statistics, and ML fundamentals • Strong Python programming skills, with C++ experience considered highly beneficial • Experience using frameworks such as PyTorch, TensorFlow, or JAX • Experience building scalable ML systems for production environments • Familiarity with cloud or GPU based training infrastructure and MLOps workflows • Experience designing evaluation, testing, and benchmarking ...

equivalent experience in Computer Science, Maths, Machine Learning, Data Science, or related field. Desirable: Exposure to machine learning frameworks such as PyTorch, JAX, Triton, TensorFlow Experience with distributed workload management systems such as Kubernetes, VLLM, Keras or MLOps pipelines Experience working with hardware simulators or emulators (e.g. QEMU). Experience ...

Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research If this sounds like ...

Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research ...

message types, and topic debugging* Experience working with physical robot hardware, camera streams, joint states, or latency-sensitive control paths* OpenCV, NumPy, pandas, PyTorch, JAX, cloud, HPC, or LeRobot dataset format experience* Experience working closely with ML researchers or robotics engineers in a fast-moving technical environment The Company ...

knowledge of ML evaluation, validation strategy, and failure modes including error analysis and iteration planning ● Experience with model development using PyTorch/TensorFlow/JAX ● Ability to communicate clearly across technical + scientific stakeholders ● Strong experience leading or coordinating cross-functional teams, and thinking strategically across product and science (prioritisation ...

About Graphcore At Graphcore, we’re building the future of AI compute.We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software ...

inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads with … Required Skills Strong experience with GCP (Compute Engine, GKE, Storage, Networking) Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF) Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes) Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM) Proficiency in Python, Bash ...

inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads with … Required Skills Strong experience with GCP (Compute Engine, GKE, Storage, Networking) Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF) Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes) Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM) Proficiency in Python, Bash ...