76 to 100 of 104 Google JAX Jobs in London

EMNLP, NAACL, ICLR) ) with specific focus on training data curation, synthetic data generation, etc.Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ...)Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ...).Excellent communication skills to report and present … open-source libraries or writing production code.Experience training large-scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud.You will enjoy:Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

EMNLP, NAACL, ICLR) ) with specific focus on training data curation, synthetic data generation, etc.* Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ...)* Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ...).* Excellent communication skills to report … source libraries or writing production code.* Experience training large-scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud.*You will enjoy:** Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

Olix Computing seeks an Architect/Staff Systems Software Engineer to own the DX-1 runtime and serving stack, connecting PyTorch/JAX to the hardware and enabling production inference at rack scale. This is a whole-stack role at the intersection of runtime, network, and accelerator with close ...

decisions Breadth across model architectures — transformers, diffusion, Mamba and others; the field hasn't converged and curiosity matters Fluency in Python and PyTorch or JAX Systems-level thinking: reasons from first principles, not from defaults Fully fluent in English — written and verbal, hard requirement Must be based in Europe ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud ...

better outcomes, 80% less admin work Raised a $30M Series A from top investors Built a team of 40+ AI experts from Tesla, Google DeepMind, NVIDIA, and Databricks At Brain Co., your work will be deployed in the real world, not stuck in research. We move fast, with more … high level familiarity with the architecture and operation of large language models. Have personally implemented models in common ML frameworks such as PyTorch, Jax or TensorFlow. Possess a strong foundation in data structures, algorithms, and software engineering principles. Exhibit excellent problem‐solving and analytical skills, with a proactive approach ...

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

learning or equivalent practical experience. Experience with training generative models (Large Language Model (LLM), image, video). Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience conducting research. Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR). Preferred qualifications: Experience with large-scale data … pipelines. Experience with training large-scale models. Proven experience conducting research in industry About the job 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 the freedom to emphasize specific types ...

learning or equivalent practical experience. Experience with training generative models (Large Language Model (LLM), image, video). Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience conducting research. Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR). Preferred qualifications Experience with large-scale data … pipelines. Experience with training large-scale models. Proven experience conducting research in industry About the job 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 the freedom to emphasize specific types ...

statistics, machine learning or equivalent practical experience.Experience with training generative models (Large Language Model (LLM), image, video).Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch).Experience conducting research.Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR).Preferred qualifications:Experience with large-scale data pipelines.Experience with training … large-scale models.Proven experience conducting research in industryAs 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 the freedom to emphasize specific types of work. As a Research Scientist, you'll setup ...

Switzerland . Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting … pipelines. Experience with training large-scale models. Cross-functional collaboration experience. Knowledge of machine learning, statistics, and diffusion model theories. About the job 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 ...

Minimum qualifications: Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting … pipelines. Experience with training large-scale models. Cross-functional collaboration experience. Knowledge of machine learning, statistics, and diffusion model theories. About the job 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 ...

scale.Minimum qualifications:Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience.Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch).Experience working in industry, working on projects from proof-of-concept through to implementation.Experience with training diffusion models.Experience conducting applied research.Preferred qualifications:Experience … inference optimization.Experience with data pipelines.Experience with training large-scale models.Cross-functional collaboration experience.Knowledge of machine learning, statistics, and diffusion model theories. 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 ...

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

practical experience. 5 years of experience developing, training, and deploying diffusion models and generative video architectures. Experience writing custom GPU/TPU kernels (e.g., JAX or PyTorch) for model serving. Preferred qualifications: PhD in Computer Science, Machine Learning, AI, or a related quantitative field or equivalent practical experience. Experience … inference. Proven track record of deploying machine learning models to production at scale or publishing widely adopted open-source research. About the job 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 ...

equivalent practical experience.5 years of experience developing, training, and deploying diffusion models and generative video architectures.Experience writing custom GPU/TPU kernels (e.g., JAX or PyTorch) for model serving.Preferred qualifications:PhD in Computer Science, Machine Learning, AI, or a related quantitative field or equivalent practical experience.Experience in frontier … optimize model performance and inference.Proven track record of deploying machine learning models to production at scale or publishing widely adopted open-source research.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 ...

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

results in top-tier conferences and journals (e.g., NeurIPS, ICML, ICLR, *ACL, EMNLP, Nature)Implementing research ideas using modern deep learning frameworks (PyTorch/JAX), model/dataset libraries (Huggingface transformers/diffusers), and efficient kernels (Triton/CUDA)Contributing to research projects on adaptive memory, latent tokenization, sparse … with CUDA kernels and/or Triton, and inference engines (vLLM, SGLang, et cetera).Experience coding with deep learning libraries such as Pytorch/JAX is essential.Fluent written and spoken English skills as well as contributions to the group culture are expected.Applicants must hold a PhD in computer science ...

Most diseases are still poorly understood at a biological level. Despite decades of research, the causal mechanisms driving many conditions remain unclear, limiting our ability to identify the right targets, design the right interventions and ...

AI ACCELERATOR Most diseases are still poorly understood at a biological level. Despite decades of research, the causal mechanisms driving many conditions remain unclear, limiting our ability to identify the right targets, design the right ...

bringExperience building ML or AI systems in productionHands-on experience with LLMs and agentic AIStrong Python, Rust, C++ or JavaExperience with PyTorch, JAX or TensorFlowExposure to LangGraph, Google ADK or similarStrong understanding of ML fundamentals and model trade-offsExperience with MLOps, multimodal AI or constrained hardware ...

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

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 stack.Build robust systems that ensure reproducible, debuggable, large-scale runs.Qualifications: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 performance issues … following would also be good to have for this role: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 datasets, or caching ...