5 of 5 JAX Jobs in Central London

experience. Strong expertise in deep learning, neural networks, and generative models (diffusion, GANs, transformers). Hands-on experience with modern ML frameworks (e.g., PyTorch, JAX, TensorFlow). Advanced programming skills in Python. Solid understanding of model training, evaluation, and scaling. Strong problem-solving, analytical, and communication skills. Ability to work ...

LLMs: Understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Proficiency in Python using PyTorch/TensorFlow/JAX framework. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP or Azure), an understanding of containerisation (e.g., Docker). Algorithms ...

understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud ...

ICLR. Hands-on experience applying deep learning models to complex, ideally temporal or financial, datasets. Expertise in Python and modern ML frameworks (TensorFlow, PyTorch, JAX, etc.). Prior experience in a commercial or industry setting post-PhD, ideally in a similarly data-driven domain. Why Apply Compensation and total rewards ...

experience strongly preferred Proven end-to-end ML experience: data pipelines, model development, deployment, and monitoring Strong Python and ML engineering skills (PyTorch or JAX) PhD in ML, Statistics, Computer Science, or related field (or equivalent research output) Ability to communicate complex ideas clearly to both technical and non-technical ...