9 of 9 JAX Jobs in the UK excluding London

LLMs: Understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Proficiency in Python using PyTorch/TensorFlow/JAX framework. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP or Azure), an understanding of containerisation (e.g., Docker). Algorithms ...

understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud ...

Effective problem-solver who takes initiative in complex production settings Experience with scientific computing, deep learning, big data, or health IT ontologies (e.g., PyTorch, JAX, Spark, HL7, FHIR) (desirable) Familiarity with cloud infrastructure (Azure/AWS), infrastructure as code, Kubernetes, Linux, Docker, data pipelines, and MLOps tools (desirable) Passion ...

large-scale datasets and/or production applications.Proficiency developing and working with large-scale models using modern machine learning packages (e.g., TensorFlow, PyTorch, Jax, Huggingface).Proficiency in building and troubleshooting modern agentic systems (prompt tuning, routing, planning, multi-agent, RAG, tool use, memory management, etc.).Experience with healthcare data, products ...

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 ...

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 ...

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/ ...

Full TimeOn-siteCambridge, UKPosted 16 hours ago60,000-120,000 GBP/Year Machine Learning Engineer job in Cambridge | Salary: 60,000-120,000Hexwired has partnered with an innovative technology company developing advanced AI and ...