7 of 7 JAX Jobs in 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 ...

outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data Terraform AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract MLOps Engineer role, outside IR35, please do apply. ...

outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data Terraform AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract MLOps Engineer role, outside IR35, please do apply click apply for full ...