19 of 19 Permanent JAX Jobs in London

have been instrumental in: building machine learning models and pipelines in Python, using common libraries and frameworks (PyTorch/CUDA, ideally with exposure to JAX, NumPy/SciPy), especially including deep learning applications; developing models for bespoke problem settings that involve high‐dimensional data (spatiotemporal, geometric, physical); iterating on network ...

Strong software engineering skills: clean code, testing, debugging, version control, and system design. Proficiency in Python with experience in ML frameworks (PyTorch, TensorFlow, or JAX). Hands‐on experience with cloud platforms (AWS, GCP, or Azure) and containerisation (Docker, Kubernetes). Solid understanding of ML fundamentals (model training, evaluation, common ...

Computer Science (or equivalent proven track record). 4 + years working on large‐scale ML codebases. Hands‐on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s). Experience in deep learning, NLP or LLMs; bonus for CUDA or data‐pipeline chops. ...

/ML incubations or complex deep‐tech cross‐functional initiatives with measurable outcomes. Expertise in Python and deep experience with deep‐learning frameworks (e.g., Jax, TensorFlow, PyTorch) and state‐of‐the‐art ML techniques. Practical experience with modern AI paradigms such as world models, foundation models, multi‐modal language models ...

Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research ...

boundaries of what AI can actually do. Experience required: Strong hands-on experience with Reinforcement Learning (RL) Python and modern ML frameworks ( PyTorch, JAX, TensorFlow ) Experience working with time-series or sensor data Ability to turn real-world problems into practical ML solutions Comfortable taking models from research into production ...

members in diverse fields Strong programming abilities in Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL Expertise with deep learning approaches such as diffusion or flow matching Proficiency in modern software development practices: code testing, documentation, packaging ...

reinforcement learning Experience with agent frameworks, tool-use planning and workflow orchestration Systems-level fluency across data pipelines, training infrastructure and inference (PyTorch, JAX, TensorFlow) Strong MLOps discipline: CI/CD for models, offline/online evaluation, safety and guardrails The ability to influence at exec level and mentor senior ...

operations research or a closely related STEM field from a top university. Strong practical experience building machine learning models in Python, ideally using PyTorch, JAX or similar frameworks. Experience with one or more of: scientific machine learning, physics-informed ML, time-series modelling, optimisation, simulation, forecasting or probabilistic modelling. Comfort ...

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 Deadline to apply ...

statistics, with coursework or projects related to information retrieval, ranking, or recommender systems. Experience with Python and common ML frameworks (e.g. PyTorch, TensorFlow, JAX) through academic, open source, or personal projects. Familiarity with evaluating model quality using offline metrics and/or A/B testing is a plus ...

work What you bring Engineering/physics degree Strong RL experience (deep RL, debugging, non‐trivial problems) Python + modern ML stack (PyTorch/JAX, NumPy, RL libs) Comfortable with time‐series sensor data Ability to turn ambiguous operational challenges into tractable ML problems Happy switching between research and practical ...

3. A working knowledge and practical experience with Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL. Key Words: Machine Learning Engineer/Data Processing/Model Building/NMR/Deep Learning/Computational Biophysics/Drug ...

customer-facing migration, including service discovery, assessment, planning, execution, and operations. Experience developing and tuning AI/ML models using frameworks such as JAX, PyTorch, or OpenXLA. Experience in one or more of the following areas: DevOps, Security, Site Reliability Engineering (SRE), data center infrastructure, networking, compute, or storage. About ...

degree or PhD in Computer Science, Engineering or a related field. Experience developing and tuning AI/ML models using frameworks such as JAX, PyTorch, or OpenXLA. Experience in one or more of the following areas: DevOps, Security, SRE. Experience in one or more of the following areas: Data Center ...

with customer‐facing migration, including service discovery, assessment, planning, execution, and operations. Experience developing and tuning AI/ML models using frameworks such as JAX, PyTorch, or Open XLA. Experience in one or more of the following areas: data center infrastructure, networking, compute, storage. Experience in one or more ...

discovery processes. Ideal candidates should possess a PhD or equivalent experience and a strong track record in ML, including practical experience with frameworks like JAX, PyTorch, or TensorFlow. #J-18808-Ljbffr ...

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

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