11 of 11 Remote Permanent JAX Jobs

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 ❯

different programming languages (e.g., Python, R, C/C++), including experience with reinforcement learning libraries in Python Experience using deep learning packages (TensorFlow, PyTorch, JAX) Experience cleaning data, prototyping, and fine-tuning state of the art deep learning models Experience with statistics, machine learning, and reinforcement learning Experience with software 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 ❯

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 ❯

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 in 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 in 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 and 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 and 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 ❯

Machine Learning Scientists/Engineers, Cambridge, UK Astex Pharmaceuticals is a global leader in innovative drug discovery, utilizing its proprietary Fragment-Based Drug Discovery platform to develop new drug candidates, with several progressing in clinical trials. The company focuses on More ❯