1 to 25 of 36 JAX Jobs in England

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 ❯

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 ❯

experience in academia or industry • We will consider various levels of experience in relevant research areas Key Skills: • Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) • Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU More ❯

machine learning, including deep learning and/or generative models. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling/ More ❯

learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling/ More ❯

learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling/ 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 ❯

Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and/ More ❯

Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and/ More ❯