76 to 93 of 93 Google JAX Jobs in the UK

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

About WRITER WRITER is where the world’s leading enterprises orchestrate AI-powered work. Our vision is to expand human capacity through superintelligence. And we’re proving it is possible – through powerful, trustworthy AI that ...

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver—to improve access to mobility … with a strong academic publication record Strong statistical/ML knowledge Proficiency in Python 3+ years of experience with modern deep learning frameworks: JAX, TensorFlow/PyTorch We prefer: Proven track record of deploying ML models to production environments Deep learning experience, especially with generative models, e.g., LLMs/ ...

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver—to improve access to mobility … generative models. Strong publication record or history of impactful project delivery in RL or related areas. Proficiency in Python and standard ML frameworks (e.g., JAX, TensorFlow). Experience with large-scale distributed training and data processing. Proven ability to lead complex and ambiguous technical projects from conception to completion. ...

problem-solving skills with the ability to work independently and collaboratively. Clear written and verbal communication skills. Familiarity with deep learning frameworks (TensorFlow, PyTorch, JAX) advantageous. Experience with optimisation, reinforcement learning, or LLMs a plus. Exposure to chip design or applied AI research desirable. Keywords: Research Engineer/Machine … Learning/AI/LLMs/Reinforcement Learning/Generative Models/Deep Learning/PyTorch/TensorFlow/JAX/Python/C++/Semiconductor/Chip Design/Optimisation/London/Cambourne/UK/Permanent If you are interested in this Research Engineer position, please ...

problem-solving skills with the ability to work independently and collaboratively. Clear written and verbal communication skills. Familiarity with deep learning frameworks (TensorFlow, PyTorch, JAX) advantageous. Experience with optimisation, reinforcement learning, or LLMs a plus. Exposure to chip design or applied AI research desirable. Keywords: Research Engineer/Machine … Learning/AI/LLMs/Reinforcement Learning/Generative Models/Deep Learning/PyTorch/TensorFlow/JAX/Python/C++/Semiconductor/Chip Design/Optimisation/London/Cambourne/UK/Permanent If you are interested in this Research Engineer position, please ...

DeepMind • London, UKBack to jobs1. Home2. Jobs3. UK4. Research Scientist, Agent Post-Training5h agoD## Research Scientist, Agent Post-TrainingDeepMind00ResearchFull-timeLondon, UKApply Job### Job DescriptionAt Google DeepMind, we've built a unique culture and work environment where long-term ambitious research can flourish. In this role, you will push … reinforcement learning. You will value various experience and backgrounds to create extraordinary impact.Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high ...

with advanced reinforcement learning topics, such as RL for sequence models, post-training, preference-based learning, or agentic systems.Familiarity with modern research stacks (e.g., JAX/Flax or PyTorch) and experience scaling experiments.Ability to be comfortable with scaling methodologies, evaluation techniques, and diagnosing complex failure modes.Ability to push projects … initiative.Strong experimental judgment, including selecting appropriate baselines and designing insightful ablations.Excellent communication skills, with a focus on clear presentation of research results.As an organization, Google maintains a portfolio of research projects driven by fundamental research, new product innovation, product contribution and infrastructure goals, while providing individuals and teams ...

have built and shipped ML systems used by real users. Examples of relevant experience include: Model training and evaluation Production inference systems PyTorch or JAX GPU-based training and inference ML experimentation and iteration Monitoring and debugging production ML systems AI reliability and performance optimisation End-to-end ownership … production Independent ownership and execution Excellent problem-solving ability A bias towards shipping, measuring outcomes, and iterating quickly Technical Environment Python PyTorch/JAX GPU-based training and inference systems OpenAI, Anthropic and open-source models SQL/NoSQL Docker Kubernetes AWS, Azure or GCP Compensation Compensation is intentionally ...

building defenses against risks, contributing to DeepMind's Frontier Safety Framework commitments. Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high … advance research in automated red-teaming and adversarial robustness, leveraging multi-turn/agentic attacks to systematically test for and uncover misuse vulnerabilities. Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users ...

hackajob is collaborating with Google to connect them with exceptional professionals for this role. At Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas may come from internal projects as well … related technical field, or equivalent practical experience. 2 years of experience with software engineering involving Python-based ML/scientific libraries such as JAX or internship experience in these areas or Open Source Software (OSS) contributions. 1 year of experience with machine learning. Preferred Qualifications Experience with large-scale ...

make that hardware serve large AI models at rack scale by building and extending the runtime and serving stack that connects PyTorch and JAX down to the metal. This is a whole-stack systems role. You’ll work where the runtime, the network, and the accelerator meet, partnering closely … cache and memory management, and reliability across distributed failure domains at cluster scale. Fluency at the framework boundary, connecting accelerators to PyTorch/JAX and serving stacks without treating either as opaque. Whole-stack debugging: end-to-end and timeline tracing, workload replay, and reasoning from architectural constraints (SRAM ...

approaches, and lessons learned You May Be a Good Fit If You: Have hands‐on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large‐scale distributed systems Genuinely enjoy both research and engineering work—you'd describe your ideal split as roughly 50/… Care about the societal impacts of AI and responsible scaling Strong Candidates May Also Have: Previous experience training LLM’s or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scale Contributed to open‐source LLM frameworks (e.g., open_lm, llm‐foundry, mesh‐transformer‐jax ...

debugging approaches, and lessons learnedYou May Be a Good Fit If You:Have hands-on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large-scale distributed systemsGenuinely enjoy both research and engineering work—you'd describe your ideal split as roughly 50/… research engineerCare about the societal impacts of AI and responsible scalingStrong Candidates May Also Have: Previous experience training LLM’s or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scaleContributed to open-source LLM frameworks (e.g., open_lm, llm-foundry, mesh-transformer-jax)Published ...

runs. You Might Be a Good Fit If You Have Strong engineering experience in large-scale distributed training or HPC systems. Deep familiarity with JAX internals, distributed training libraries, or custom kernels/fused ops. Experience with multi-node cluster orchestration (Slurm, Ray, Kubernetes, or similar). Comfort debugging … closely with infra, research, and deployment teams. Nice to Have Experience with training LLMs or other large transformer architectures. Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.). Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches). Experience with data pipeline optimization, sharded ...

systems Strong Python software engineering skills Experience training, fine-tuning or deploying modern machine learning models Strong experience with PyTorch and/or JAX Experience building scalable ML infrastructure and inference pipelines Comfortable owning projects end-to-end in fast-moving environments Previous mentoring or technical leadership experience would … advantageous Tech Stack Python * PyTorch * JAX * GPU Training * LLMs * Modern ML Infrastructure Get in touch for more details - McGregor Boyall is an equal opportunity employer and do not discriminate on any grounds. ...

Member of Technical Staff, Training Infrastructure — Inherent (London) At Inherent, we are on a mission to build AI that recursively self‐improves to discover new knowledge. Scientific advances are the backbone of our economic, technological ...

NLP PEOPLE is looking for a Machine Learning Engineer to build and deploy advanced models to optimize complex operational systems. The role involves hands-on engineering, working closely with the founding team and customer data. ...

Bjak is seeking a Technical Lead in Machine Learning to oversee the execution layer of their AI chat app. This role involves managing ML systems, ensuring they are trainable and deployable under real-world constraints. ...