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

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

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 architectures) Desired ...

modern ML models (e.g. transformers, diffusion, graph models, sequence models). Solid software skills in Python and at least one major ML framework (PyTorch, JAX, or TensorFlow). Experience designing and running non-trivial experiments: controlling for confounders, building robust baselines, and doing thorough error analysis. Ability to write clearly ...

. Domain Depth: Hands-on expertise in VLMs , Diffusion/Generative Video , or LLM Pre-training . Frameworks: Mastery of Python and PyTorch or JAX . Systems: Experience with distributed training and complex numerical debugging. Randstad Technologies Ltd is a leading specialist recruitment business for the IT & Engineering industries. Please ...

. Domain Depth: Hands-on expertise in VLMs , Diffusion/Generative Video , or LLM Pre-training . Frameworks: Mastery of Python and PyTorch or JAX . Systems: Experience with distributed training and complex numerical debugging. Randstad Technologies Ltd is a leading specialist recruitment business for the IT & Engineering industries. Please ...

classifiers Behaviour-shaping or reward-modelling components. LLM and multimodal fine-tunes Adversarial robustness defences Use Python and modern ML frameworks (e.g., PyTorch, TensorFlow, JAX, HuggingFace). Contribute to creation of synthetic datasets, adversarial evaluation corpora, and scenario-based test sets. Help transition research outputs into scalable controls for engineering … Responsible AI concepts: Bias and fairness Explainability Privacy-preserving ML. Robustness and uncertainty Nice to have: Experience with ML frameworks: PyTorch, TensorFlow, Flax/JAX, HuggingFace. Exposure to multimodal models (CLIP, Whisper, LLaVA, video transformers). Familiarity with safety benchmarks, adversarial testing, red teaming, or uncertainty estimation. Knowledge ...

with teleops teams to transform human trajectories into robust robotic skills through behavior cloning. High-Performance Engineering: Designing and profiling research-grade PyTorch/JAX code to support large-scale, distributed RL infrastructure. Essential Skills Needed Deep Learning Mastery: 5+ years building and shipping models, with deep hands-on expertise … Expert: A proven track record of solving complex, real-world problems using Deep Reinforcement Learning. Technical Rigour: Mastery of Python and PyTorch/JAX, including the ability to profile performance and debug complex numerical stability issues. Robotics Foundation: Practical experience with simulators (Isaac Sim/MuJoCo) and a deep understanding ...

with teleops teams to transform human trajectories into robust robotic skills through behavior cloning. High-Performance Engineering: Designing and profiling research-grade PyTorch/JAX code to support large-scale, distributed RL infrastructure. Essential Skills Needed Deep Learning Mastery: 5+ years building and shipping models, with deep hands-on expertise … Expert: A proven track record of solving complex, real-world problems using Deep Reinforcement Learning. Technical Rigour: Mastery of Python and PyTorch/JAX, including the ability to profile performance and debug complex numerical stability issues. Robotics Foundation: Practical experience with simulators (Isaac Sim/MuJoCo) and a deep understanding ...

Deep Learning Engineer About the Company Join a fast-growing AI-driven technology company that’s modernising prediction and decision systems for complex, real-world industries. The organisation builds powerful AI platforms that help partners ...