201 to 225 of 267 Google JAX Jobs in England

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

Staff Software Engineer, GPU Performance, Core ML link Copy link corporate_fare Google place London, UK Advanced Experience owning outcomes and decision making, solving ambiguous problems and influencing stakeholders;deep expertise in domain. Apply link Copy link Bachelor's degree or equivalent practical experience. 8 years of experience in software development and with data structures/algorithms (e.g., C++ … modern GPU architectures, memory hierarchies, and performance bottlenecks. Ability to develop and utilize sophisticated performance models and benchmarks to guide optimization efforts and hardware roadmap decisions. About the job Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information … natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities 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 ❯

/EE or related research 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) • Experience with distributed and 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 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 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 More ❯

skills — Python, REST APIs, FastAPI 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 rapidly rising valuation — early More ❯

skills — Python, REST APIs, FastAPI 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 rapidly rising valuation — early More ❯

skills — Python, REST APIs, FastAPI 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 rapidly rising valuation — early More ❯

skills — Python, REST APIs, FastAPI 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 rapidly rising valuation — early 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 ❯

field (or equivalent practical experience). 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 integrated hardware-software systems. More ❯

field (or equivalent practical experience). 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 integrated hardware-software systems. More ❯

field (or equivalent practical experience). 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 integrated hardware-software systems. 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 ❯

At UnlikelyAI, we're looking for a visionary Staff Applied Scientist to play a key leadership role in our Applied Science team and across the company. This is a high-impact individual contributor position with mentoring and coaching responsibilities - ideal More ❯