201 to 225 of 246 JAX Jobs in England

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 ❯

At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯

At UnlikelyAI, we're looking for a visionary Staff Applied Scientist to play a key leadership role in our Applied Science team and across the company. This is a high-impact individual contributor position with mentoring and coaching responsibilities - ideal More ❯

our London office (66 City Rd). Your Focus Design and implement robust experiments to evaluate specific model capabilities Build and maintain high-frequency evaluation pipelines using PyTorch or JAX Engineer benchmark datasets — collecting, filtering, and decontaminating data for meaningful evals Create evaluation protocols that measure the right capabilities and avoid metric gaming Research and implement strong baselines from literature … infra and tooling About You Proficient hands-on experience in machine learning, ideally with a focus on experimental design or evaluation Strong engineering skills in Python and PyTorch (or JAX) Deep understanding of training and evaluating large-scale deep learning models A scientific mindset — you know how to design a clean experiment and what makes a result trustworthy Comfortable building More ❯

our London office (66 City Rd). Your Focus Design and implement robust experiments to evaluate specific model capabilities Build and maintain high-frequency evaluation pipelines using PyTorch or JAX Engineer benchmark datasets — collecting, filtering, and decontaminating data for meaningful evals Create evaluation protocols that measure the right capabilities and avoid metric gaming Research and implement strong baselines from literature … infra and tooling About You Proficient hands-on experience in machine learning, ideally with a focus on experimental design or evaluation Strong engineering skills in Python and PyTorch (or JAX) Deep understanding of training and evaluating large-scale deep learning models A scientific mindset — you know how to design a clean experiment and what makes a result trustworthy Comfortable building More ❯

skills — Python, REST APIs, FastAPI Proven experience scaling cloud-native systems (AWS, microservices, IaC) Kubernetes + Terraform expertise in production environments Solid understanding of ML deployment (e.g. PyTorch, TensorFlow, JAX) Frontend skills with React/TypeScript or similar Bonus: big data/real-time analytics/multi-cloud Why Join? Equity with serious upside and rapidly rising valuation — early-stage More ❯

skills — Python, REST APIs, FastAPI Proven experience scaling cloud-native systems (AWS, microservices, IaC) Kubernetes + Terraform expertise in production environments Solid understanding of ML deployment (e.g. PyTorch, TensorFlow, JAX) Frontend skills with React/TypeScript or similar Bonus: big data/real-time analytics/multi-cloud Why Join? Equity with serious upside and rapidly rising valuation — early-stage More ❯

You may be a good fit if you: Have 5+ years of industry-related experience Are proficient in Python and experienced with deep learning frameworks such as PyTorch or Jax Possess a strong software engineering background and enjoy working closely with researchers and engineers Enjoy pair programming Care about code quality, testing, and performance Are passionate about AI's potential More ❯

on clusters/cloud. Familiarity with modern scientific software practices (version control, containerisation, continuous integration etc) and an enthusiasm for writing clear, performant code. Familiarity with ML frameworks (PyTorch, JAX or similar) and machine-learned potential energy surfaces. Previous computational drug discovery experience in industry. If you do not quite fulfil all of the requirements or are not sure which More ❯

related field. Demonstrated success shipping LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech, pharma, fintech, etc.). Strong More ❯

related field. Demonstrated success shipping LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech, pharma, fintech, etc.). Strong More ❯

related field. Demonstrated success shipping LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech, pharma, fintech, etc.). Strong More ❯

all trades” in MLSys. Skills and Qualifications MS or PhD in CS/EE or equivalent experience in the industry, with key skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantisation, pruning, sparsification, etc.) Experience with deployment on embedded and mobile devices (ML inference and/or training) Experience with distributed and multi-GPU training at More ❯

experience with modern Machine Learning Ops lifecycle, tools, and frameworks Nice to have: Experience supporting machine learning research activities Familiarity with python-based ML frameworks like TensorFlow, PyTorch, or JAX Experience with Google Cloud Platform (GCP) Culture and values What does it take to be successful at IsoLabs? It's not about finding people who think and act in the More ❯

ideally including transformers, diffusion models, or multimodal architectures. Experience applying ML to large multimodal biological datasets (genomics, transcriptomics, epigenomics, proteomics, single-cell data). Exceptional coding skills (Python, PyTorch, JAX, TensorFlow) and familiarity with scalable training frameworks. Experience managing large-scale distributed training across GPUs/TPUs. Background in generative AI, large language models, mechanistic interpretability, or reinforcement learning. Interview More ❯

Relevant research experience to the position such as post doctoral roles, a proven track record of publications, or contributions to machine learning codebases. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas Strong knowledge of linear algebra, calculus and statistics Experience working in a scientific environment across disciplines (particularly biology More ❯

Relevant research experience to the position such as post doctoral roles, a proven track record of publications, or contributions to machine learning codebases. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas Strong knowledge of linear algebra, calculus and statistics Experience working in a scientific environment across disciplines (particularly biology More ❯

solving skills.Good understanding of machine learning fundamentals.GPU compute programming models & optimization techniques.GPU compute framework development, maintenance, and optimization.Machine learning development using one or more ML frameworks (TensorFlow, PyTorch or JAX).Experience with adding computational graph support, runtime or device backend to Machine learning libraries (TensorFlow, PyTorch or JAX) support is a plus.Experience with high performance parallel programming, GPU programming or More ❯

choice of loss functions, regularization techniques, model architecture, optimizer, learning rate schedules, etc. Thorough understanding of Python tools and libraries, keen to offer advice on best practices Experience using Jax, Tensorflow or similar ML frameworks a huge plus Benefits: Market-leading salaries Generous benefits package, including physical & mental health benefits, excellent holiday entitlement, significant parental leave, retirement benefits, private on More ❯

choice of loss functions, regularization techniques, model architecture, optimizer, learning rate schedules, etc. Thorough understanding of Python tools and libraries, keen to offer advice on best practices Experience using Jax, Tensorflow or similar ML frameworks a huge plus Benefits: Market-leading salaries Generous benefits package, including physical & mental health benefits, excellent holiday entitlement, significant parental leave, retirement benefits, private on More ❯

of commercial products An understanding of the latest trends in AI/ML research and development Good familiarity with one of the popular ML frameworks such as PyTorch, TensorFlow, JAX, ONNX, or PaddlePaddle A history of developer engagement or participation in an active open-source development community Experience in Product Owner roles, or a desire to grow into more commercial More ❯

choice of loss functions, regularization techniques, model architecture, optimizer, learning rate schedules Thorough understanding of Python tools and libraries, with the ability to advise on best practices Experience with Jax, TensorFlow, or similar ML frameworks is a plus Benefits: Market-leading salaries Generous benefits including health, mental health, holiday entitlement, parental leave, retirement plans, private on-site gym Focus on More ❯

latency ML pipelines in high-performance environments is a strong plus Strong engineering skills, including Python, CUDA, or C++ Knowledge of machine learning frameworks such as PyTorch, TensorFlow, or JAX Proficiency in GPU programming for training and inference acceleration (e.g., CuDNN, TensorRT) Experience with distributed training for scaling ML workloads (e.g., Horovod, NCCL) Exposure to cloud platforms and orchestration tools More ❯

systems (AI) & reinforcement learning) Comfort 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 a fast-paced environment—ideal More ❯

and requirements for the autonomous driving stack. Experience MS or higher degree in a related field, or equivalent industry experience. Professional experience with ML frameworks such as PyTorch (preferred), Jax, or TensorFlow. Knowledge of debugging and profiling neural networks on NVIDIA CUDA. Experience with architectures for object detection and 3D perception. Experience in ML workflows: data sampling, pre-processing, training More ❯