76 to 100 of 110 JAX Jobs in London

lead complex research efforts while staying grounded in product impact, user experience, and real-world performance. Strong coding and experimentation skills (Python, PyTorch/JAX, audio processing libraries). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Proven experience mentoring others ...

integrating ML systems into production while staying grounded in product impact and real-world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

evaluation, and testing Optimise reliability, scalability, latency, and user experience Work closely with Product and Engineering teams to solve complex customer problems Python PyTorch JAX vLLM Vector Databases What We're Looking For Strong software engineering fundamentals Experience building AI agents, copilots, RAG systems, or workflow automation platforms Excellent Python ...

Languages* Machine Learning Models* Model Training* Engineering Workflows* Data Analysis at Scale* Data Visualization* Data Ingestion Pipelines* Large Language Models (LLMs)* AI Agents* Tools* JAX* PyTorch #J-18808-Ljbffr ...

integrating ML systems into production while staying grounded in product impact and real‐world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

models, fighting deep learning compilers). Excellent understanding of GPU architecture and why some models run faster than others. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice to have: Robotics or autonomous ...

impacting real world drug design programs.Skills and qualifications Essential:Significant experience with large scale distributed training of LLMs.Experience with deep learning ML frameworks (either JAX or PyTorch).Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL).Good understanding of GPU architectures. Reasoning about performance concepts is more important than ...

data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

from data to UI. Applicants should have strong quantitative skills and experience in Python/SQL/TypeScript and PyData stack, plus PyTorch/JAX/TF for model development. #J-18808-Ljbffr ...

live training/matchday schedule, not just be technically neat. Python/SQL/TypeScript, PyData stack (Pandas/Polars, Altair, etc.), PyTorch/JAX/TF. You'll own things end-to-end, data through UI. Looking for strong quant background, first-principles engineering judgment, and taste ...

high agency and low ego Strong candidates may also have Experience with large-scale distributed training (RL, pre-training, or post-training) Familiarity with JAX, PyTorch, or similar ML frameworks A track record of operating at the edge of research and infra in a fast-moving environment Compensation Annual Salary ...

have strong experience building production-grade machine learning systems at scale You are experienced with modern ML frameworks such as PyTorch, TensorFlow, or JAX You have worked with or are interested in multimodal machine learning You understand how to design systems that balance automation with quality, safety, and user experience ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

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

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

Req ID:365825NTT DATA strives to hire exceptional, innovative and passionate individuals who want to grow with us. If you want to be part of an inclusive, adaptable, and forward-thinking organization, apply now.We are ...

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

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

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

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

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 … London, UK; Zürich, 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 applied research. ...

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/50 rather … 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 research ...

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 AI research … 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. ...

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