17 of 17 JAX Jobs in the South East

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

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

system design Required Experience * Strong software engineering background with Python and or C++* Hands-on experience with ML frameworks such as PyTorch, TensorFlow, or JAX* Experience building or maintaining ML infrastructure, training platforms, or inference systems* Solid understanding of distributed systems, parallel computing, and large-scale data processing* Strong fundamentals ...

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

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

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

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

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

Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join ...

engineers to ship fast in an environment that iterates constantly What they're looking for Strong Python engineering background Experience with frameworks like PyTorch, JAX or TensorFlow Demonstrated ability to scale ML workflows in real-world settings (cloud, GPU clusters, distributed training) Comfortable with MLOps tooling (W&B, Ray, Docker ...

AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise ...