16 of 16 Permanent JAX Jobs in London

pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling ...

pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling ...

pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling ...

pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling ...

forests, and ensemble methods Experience with generative models (GANs, VAEs, Diffusion models) Programming & Tools : Expert-level Python programming Proficiency with ML frameworks (PyTorch, TensorFlow, JAX) Experience with scikit-learn, XGBoost, LightGBM Strong software engineering practices and clean code principles Data & Computing : Experience with big data technologies (Spark, Dask ...

forests, and ensemble methods Experience with generative models (GANs, VAEs, Diffusion models) Programming & Tools : Expert-level Python programming Proficiency with ML frameworks (PyTorch, TensorFlow, JAX) Experience with scikit-learn, XGBoost, LightGBM Strong software engineering practices and clean code principles Data & Computing : Experience with big data technologies (Spark, Dask ...

modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs ...

modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs ...

software and research teams. Requirements Strong background in machine learning algorithm development and optimisation. Proficiency in Python and modern ML frameworks such as PyTorch, JAX or TensorFlow. Experience with data modelling, preprocessing and performance benchmarking. Understanding of ML evaluation metrics and model analysis methods. Excellent communication skills and collaborative mindset. ...

software and research teams. Requirements Strong background in machine learning algorithm development and optimisation. Proficiency in Python and modern ML frameworks such as PyTorch, JAX or TensorFlow. Experience with data modelling, preprocessing and performance benchmarking. Understanding of ML evaluation metrics and model analysis methods. Excellent communication skills and collaborative mindset. ...

transformers, diffusion, retrieval-augmented systems, reinforcement learning, etc.). Strong coding skills in Python and experience with at least one major ML framework (PyTorch, JAX, TensorFlow). Ability to bridge high-level research with practical implementation. Curious, humble, and excited to work on hard technical problems with real enterprise impact. ...

transformers, diffusion, retrieval-augmented systems, reinforcement learning, etc.). Strong coding skills in Python and experience with at least one major ML framework (PyTorch, JAX, TensorFlow). Ability to bridge high-level research with practical implementation. Curious, humble, and excited to work on hard technical problems with real enterprise impact. ...

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. 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 ...

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. 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 ...

models and generative AI Proven experience with text-to-image generation, multimodal AI Large-scale model training experience (distributed systems, GPUs) Proficiency in PyTorch, JAX, or similar frameworks Enterprise AI deployment experience Safety and alignment research background Multilingual/multicultural AI expertise Highly Valued: A Background at top-tier ...

models and generative AI Proven experience with text-to-image generation, multimodal AI Large-scale model training experience (distributed systems, GPUs) Proficiency in PyTorch, JAX, or similar frameworks Enterprise AI deployment experience Safety and alignment research background Multilingual/multicultural AI expertise Highly Valued: A Background at top-tier ...