3 of 3 Permanent JAX Jobs in Oxford

practices.Experience building machine-learning, generative-AI or LLM applications, gained through employment, industry or research internships, applied research or significant academic projects.Experience with PyTorch, JAX, TensorFlow or a similar framework.Familiarity with core agentic AI concepts such as tool calling, planning, memory and workflow orchestration.An understanding of reinforcement learning, preference optimisation ...

silicon. What they’re looking for: Degree in Physics/Engineering/Electronics and a PhD in Machine Learning Strong Python + PyTorch/JAX skills Publication record in ML or EDA venues Ability to move between research and implementation Clear communication across ML and circuit‐design audiences Experience modelling ...

silicon. What they're looking for: First degree in relevant STEM subject 3+ years in industry building production ML systems Python + PyTorch/JAX expertise for training, inference, and model export Experience building data pipelines for complex scientific/engineering datasets Solid software engineering fundamentals - testing, CI/ ...