26 to 50 of 97 JAX Jobs in London

Legal, Compliance, and external partners like Google DeepMind.Skills and qualifications EssentialAI/ML Engineering: Deep conceptual and practical understanding of deep learning frameworks (e.g., JAX, PyTorch, TensorFlow), large-scale cloud training or inference infrastructure, and LLM ecosystems.Adversarial ML Mindset: Strong familiarity with AI security threat vectors (prompt injection, model inversion ...

prototypes to reliable deployment; Have experience partnering closely with researchers throughout implementation; Be proficient in Python and deep learning frameworks such as PyTorch or JAX; Have strong software engineering fundamentals, including clean, testable, well‐documented and maintainable code, version control and code reviews; Have experience with distributed training frameworks such ...

large, possibly noisy, data sets. Strong analytical and quantitative skills, and a detail-oriented mindset. Strong proficiency in machine learning libraries such as Torch, JAX or TensorFlow. Competence in Python, cluster environment, and general software engineering principles (source control, testing, collaborative workflow). Excellent written and verbal communication skills, willing ...

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play a key role ...

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

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch Distributed ...

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch Distributed ...

performance engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems.Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies.Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live quantitative strategies.Experience ...

strong programming skills in Python , Rust, or similar; and strong software engineering fundamentals building performant and reliable systems. Proficient in deep learning frameworks (Pytorch, JAX, TensorFlow). You can work on different layers of the stack from low-level training backends, data ingestion to ML/RL algorithmic design ...

record of shipping products and features inside a fast-moving environment Strong coding skills in Python and familiarity with ML frameworks like PyTorch or JAX Experience training, fine-tuning, and evaluating large language models Deep familiarity with large language models and web agents, how they work, and how they ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

especially knowledge in formal methods, stochastic systems and processes, the foundations of deep learning. Experience coding with deep learning libraries such as Pytorch/JAX is essential. Applicants must hold a PhD in computer science, mathematics or equivalent experience.Please see job description for a full list of requirements. What ...

series). Experience limited to API usage or prompt engineering is not sufficient.Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks.Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking.Evidence of research/technical quality ...

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

agents, LLM research, as well as reinforcement learning and fine-tuning Strong knowledge of linear algebra, calculus and statistics.Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas.Depending on your experience: project supervision, leadership, or management.Strong knowledge of LLM landscape today ...

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 is a bonus rather ...

creative teams to turn ML capabilities into compelling experiences What We’re Looking For: Strong knowledge of Python Experience writing models in PyTorch or JAX A good understanding of how large language models work Comfortable working along the whole ML pipeline: from data preparation to model design, training and evaluation ...

optimizing model serving infrastructure or inference backends. Preferred qualifications: Experience with developing serving infrastructure. Experience programming hardware accelerators (GPUs, TPUs) via ML frameworks (e.g., JAX, PyTorch) or low-level programming models (e.g., Pallas, CUDA, OpenCL). Experience profiling software to identify performance bottlenecks. Experience with distributed ML systems optimization ...

About You Strong Python engineering skills and experience working in large codebases (testing, code review, CI, operational ownership). Hands‐on experience with PyTorch, JAX, or similar. Strong systems and infrastructure fundamentals. Experience with LLM training or post‐training: fine‐tuning, RL, distillation, evaluation, and/or data pipelines. Excellent ...

adapting, or evaluating large-scale foundation models, and building data pipelines for heterogeneous datasets. 3 years of experience with modern machine learning frameworks (e.g., JAX, PyTorch, TensorFlow) and distributed training on accelerators. 3 years of experience designing evaluations, metrics, and controlled ablations for research projects. Preferred qualifications: Master's degree ...

refine existing products.Prior experience in working cross-discipline in matrixed organizations. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.Experience using large-scale distributed training and inference.Strong mentorship, communication, and problem-solving skills.A demonstrated passion for applied ML models and products.Bonus: experience ...

refine existing products. Prior experience in working cross-discipline in matrixed organizations. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.Experience using large-scale distributed training and inference. Strong mentorship, communication, and problem-solving skills. A demonstrated passion for applied ML models ...

high agency and low egoStrong candidates may also haveExperience with large-scale distributed training (RL, pre-training, or post-training)Familiarity with JAX, PyTorch, or similar ML frameworksA track record of operating at the edge of research and infra in a fast-moving environmentDeadline to apply: None. Applications will ...

learning techniques, including designing new architectures, hands-on experimentation, analysis, and visualisation.Strong knowledge of linear algebra, calculus and statistics.Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas.A passion for applying ML research to real world problems.Depending on your experience: project ...