126 to 144 of 144 Google JAX Jobs in the UK excluding London

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, york col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry 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/ 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 More ❯

responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques such as quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning frameworks like PyTorch. Experience building Deep Reinforcement Learning models. Experience with robotics More ❯

experience, with a strong record of transitioning research into production. Proven team leadership in a technical or research setting. Expert in Python and PyTorch; JAX is a plus. Strong foundation in data preparation, feature engineering, and model evaluation. Excellent communicator with the ability to adapt messaging for different audiences. More ❯

Experience shipping real products to customers ML Design and Deployment: Familiarity with designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT, HuggingFace Transformers, or commercial APIs. Containerisation: Experience with cloud deployments and containerisation (Docker, Kubernetes, containerd). Linux: Embedded Linux and kernel development. Programming 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 ❯

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

role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search … LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with … AI/ML frameworks (e.g., TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning More ❯

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

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: Daxtra is at the forefront of AI-driven innovation, leveraging cutting-edge technologies to develop intelligent solutions that enhance business outcomes. We are looking for an AI Engineer to work closely with our AI/ML team, contributing 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 ❯

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 ❯

Compiler Engineers to join the team building the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs that connect PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers.Your mission will be to create seamless support for a broad ecosystem of large AI models … impact technical and business milestones that drive the success of Flux. Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware. Optimise Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput. Mentor & Encourage Standards: Lead … C/C++ programming. Deep expertise in ML framework internals, compilers, low-level programming, and optimisation techniques. Deep expertise in optimising Tensorflow, PyTorch or JAX deep learning models. Deep expertise with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM. Practical experience applying machine learning in high-performance computing More ❯

to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI … ensure software and hardware are designed together to deliver maximum performance. Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware. Optimize Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximize throughput. Mentor & Define Standards: Lead … on C/C++ programming. Extensive experience in ML framework internals, compilers, low-level programming, and optimisation techniques. Extensive experience optimising Tensorflow, PyTorch or JAX deep learning models. Extensive experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM. Practical experience applying machine learning in high-performance computing More ❯

Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯