JAX Jobs in the Thames Valley

5 of 5 JAX Jobs in the Thames Valley

AI Intern

slough, south east england, United Kingdom
Hybrid / WFH Options
Capua
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 ❯
Posted:

Machine Learning Engineer (Visual Specialist)

slough, south east england, United Kingdom
Hybrid / WFH Options
OhChat
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 ❯
Posted:

Postdoctoral Researcher (m/f/x) in Machine Learning and Artificial Intelligence

Oxford, Oxfordshire, United Kingdom
Plattnerfoundation
spoken and written) Strong writing, presentation, and analytical skills Research experience with publications appropriate to career stage Experience with ML frameworks like PyTorch or Jax Self-motivated, able to work independently and meet deadlines Willing to support teaching and mentor students Interest in organizing academic events Desirable qualifications include experience More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Head of Applied AI

slough, south east england, United Kingdom
Hybrid / WFH Options
Acquired Talent Ltd
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 ❯
Posted:

Artificial Intelligence Engineer

slough, south east england, United Kingdom
Understanding Recruitment
AI Engineer (London) A fast-growing defence tech startup focused on AI and software for autonomous systems is hiring an AI Research Engineer for their remote team. Partnered with a third-party recruiter, this well-funded company is committed to More ❯
Posted: