1 to 25 of 26 JAX Jobs in London

MAS). Advanced Python programming skills, with a strong emphasis on writing efficient, scalable, and maintainable code. Proven experience with TensorFlow/PyTorch/Jax, Scikit-learn, and MLOps workflows for training, deployment, and monitoring of ML models. Experience working with Polars and/or Pandas for high-performance data More ❯

or a related field, or equivalent professional experience Strong programming skills in Python, with experience in machine learning frameworks such as PyTorch, TensorFlow or JAX Hands-on experience fine-tuning LLMs and optimising hyperparameters for improved performance Skilled in building LLM-powered applications for real-time decision-making and automation More ❯

advanced machine learning techniques, with a focus on LLMs/genAI, optimisation, or sequential learning Strong background in Python, deep learning frameworks (TensorFlow, PyTorch, Jax), and ML model development PhD in Computer Science, Machine Learning, Applied Mathematics, or related field, or equivalent experience A history of successful scientific publications and More ❯

advanced machine learning techniques, with a focus on LLMs/genAI, optimisation, or sequential learning Strong background in Python, deep learning frameworks (TensorFlow, PyTorch, Jax), and ML model development PhD in Computer Science, Machine Learning, Applied Mathematics, or related field, or equivalent experience A history of successful scientific publications and More ❯

advanced machine learning techniques, with a focus on LLMs/genAI, optimisation, or sequential learning Strong background in Python, deep learning frameworks (TensorFlow, PyTorch, Jax), and ML model development PhD in Computer Science, Machine Learning, Applied Mathematics, or related field, or equivalent experience A history of successful scientific publications and More ❯

record in top-tier conferences (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR). Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ). Excellent communication skills to report and present research More ❯

Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry-specific More ❯

regression techniques, etc.) Familiarity with Deep Learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) Familiarity with PyTorch , TensorFlow , JAX or similar frameworks Strong coding abilities in Python and/or C++ Preferred Qualifications: 2D & 3D Generative AI Reinforcement Learning LLMs and Natural Language Processing More ❯

data-driven role, with exposure to: building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); iterating on network architectures and model More ❯

quality ML systems. Preferred Qualifications: Master's degree or PhD in Computer Science or a related technical field. Experience with modern ML frameworks (e.g., JAX, Pytorch or TensorFlow). Experience with launching applied Machine Learning/Natural Language Processing (ML/NLP) projects. About the Job Google's software engineers More ❯

machine learning, with 2+ years of hands-on experience in LLM fine-tuning, RAG systems, and prompt engineering. Proficiency with frameworks like PyTorch, Tensorflow, JAX, Pandas, and OpenCV. A solid understanding of transformer models and their common variants, loss functions, and pitfalls. A quick learner with a structured, organized approach More ❯

by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong More ❯

applications to engineering teams and C-Level executives. This requires deep familiarity across the stack - compute infrastructure (Amazon EC2, Amazon EKA), ML frameworks PyTorch, JAX, orchestration layers Kubernetes and Slurm, parallel computing (NCCL, MPI), MLOPs, through to Amazon SageMaker Hyperpod, Amazon Bedrock as well as target use cases in the More ❯

conferences. Expertise in Programming: Strong proficiency in software development and programming in Python. Experience with Deep Learning Frameworks: Proficiency in frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent Communication Skills: Ability to clearly communicate complex ideas, both verbally and in writing, and to work effectively within a multidisciplinary More ❯

conferences. Expertise in Programming: Strong proficiency in software development and programming in Python. Experience with Deep Learning Frameworks: Proficiency in frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent Communication Skills: Ability to clearly communicate complex ideas, both verbally and in writing, and to work effectively within a multidisciplinary More ❯

conferences. Expertise in Programming: Strong proficiency in software development and programming in Python. Experience with Deep Learning Frameworks: Proficiency in frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent Communication Skills: Ability to clearly communicate complex ideas, both verbally and in writing, and to work effectively within a multidisciplinary More ❯

to-end stack; Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for end More ❯

and deploying GenAI models and applications on cloud infrastructure. Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. Knowledge of MLOps tools and workflows for model development, validation, and deployment. High level of communicating effectively across internal and external organizations, with More ❯

and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations, with More ❯

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯

field. You may be a good fit if you have: Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray). Experience using large-scale distributed training strategies. Hands More ❯

Data Scientist, AWS Generative AI Innovation Center Job ID: Amazon Web Services EMEA Dubai FZ Branch - Q29 Amazon launched the Generative AI Innovation Center (GenAIIC) in June 2023 to help AWS customers accelerate the use of generative AI to solve More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯