26 to 34 of 34 Google JAX Jobs

that Comfortable across the full stack: distributed training frameworks, GPU/accelerator optimisation, data pipelines, experiment tracking Contributions to open-source projects like PyTorch, JAX, MLX, or similar Strong software engineering fundamentals: clean code, good testing habits, a focus on performance Nice to have Experience with biological data: genomic … like to discuss the role further. Keywords: Senior Research Engineer, Research Engineer, AI Research, Distributed Training, GPU Optimisation, CUDA, XLA, PyTorch, JAX, Machine Learning Infrastructure, Genomics, Computational Biology, Biotech, London ...

NVLink/InfiniBand), Kubernetes/Slurm, and OS internals Expertise in Python and deep experience with modern deep learning frameworks (PyTorch and/or JAX) Advanced degree (MS or PhD) in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience at a frontier … orientation, and excellent collaborative communication Reliable and predictable availability Bonus Points Expertise in: CUDA kernel development, Triton/Pallas/CuTe DSLs, PyTorch/JAX internals, XLA optimization, or hardware acceleration (FPGA/ASIC) Knowledge of reinforcement learning, post-training, or fine-tuning techniques Knowledge of financial markets ...

We are supporting a pioneering technology organisation seeking a Senior Research Engineer to lead complex AI research and engineering initiatives. This position sits at the intersection of advanced research and practical engineering. You will help ...

supervised fine-tuning (SFT), or agent-based systems, including modeling or applied infrastructure. 2 years of experience with machine learning frameworks such as JAX, Flax, TensorFlow, or PyTorch, and scaling experiments. Preferred qualifications PhD in Computer Science, Machine Learning, or a related quantitative field. Publications in reinforcement learning … artifacts. Preference for simple, elegant solutions to research problems. Experience with systems design and managing code complexity in large codebases. About the job At Google DeepMind, we've built a unique culture and work environment where long-term ambitious research can flourish. In this role, you will push ...

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

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

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

CoreWeave is the Essential Cloud for AI. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and teams that enables innovators to build and scale AI with confidence. Trusted by leading AI ...