9 of 9 Remote Google 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 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 More ❯

from a top-tier university. • Solid experience in training neural networks, tuning hyperparameters, and working with frameworks like PyTorch. • Knowledge of differentiable programming (e.g., JAX) or geometric deep learning (e.g., PyTorch Geometric) is a plus. • Strong foundation in mathematical concepts such as stochastic calculus, optimisation, and simulation techniques. • Proficiency More ❯

from a top-tier university. • Solid experience in training neural networks, tuning hyperparameters, and working with frameworks like PyTorch. • Knowledge of differentiable programming (e.g., JAX) or geometric deep learning (e.g., PyTorch Geometric) is a plus. • Strong foundation in mathematical concepts such as stochastic calculus, optimisation, and simulation techniques. • Proficiency More ❯

focus on GenAI 🚀 Strong expertise in Stable Diffusion, Flux, and similar architectures 🚀 Experience fine-tuning and optimising visual models 🚀 Hands-on experience with PyTorch, JAX, or TensorFlow 🚀 Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT) 🚀 Experience deploying ML models using Kubernetes and/or Docker 🚀 Strong experience with Python More ❯

focus on GenAI 🚀 Strong expertise in Stable Diffusion, Flux, and similar architectures 🚀 Experience fine-tuning and optimising visual models 🚀 Hands-on experience with PyTorch, JAX, or TensorFlow 🚀 Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT) 🚀 Experience deploying ML models using Kubernetes and/or Docker 🚀 Strong experience with Python 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 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 More ❯

With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select … 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. More ❯