7 of 7 JAX Jobs in the South East

Infrastructure as Code, and DevOps practices Production support, monitoring, observability, and incident management Experience with AI/ML frameworks such as PyTorch, TensorFlow, or JAX Exposure to LLMs, RAG, Agentic AI, model serving, or MLOps environments What's on Offer? Work on innovative AI and Agentic AI solutions Hybrid working ...

equivalent practical experience. Strong Python programming and software‐engineering skills. Experience developing machine‐learning, generative AI or LLM applications. Practical experience with PyTorch, JAX, TensorFlow or a similar framework. Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration. Knowledge of reinforcement learning, preference optimisation ...

related discipline, or equivalent practical experience.Strong Python programming and software-engineering skills.Experience developing machine-learning, generative AI or LLM applications.Practical experience with PyTorch, JAX, TensorFlow or a similar framework.Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration.Knowledge of reinforcement learning, preference optimisation or sequential decision ...

experience, with at least 2 years in an ML or AI-adjacent role Strong Python skills and familiarity with major ML frameworks (PyTorch or JAX); comfortable reading and modifying model code Hands-on experience building and operating ML pipelines in production: data pipelines, training orchestration, evaluation, and serving Experience with ...

language models, retrieval systems and domain-specific tools. Qualifications include a degree in CS or related field, strong Python, and experience with PyTorch/JAX/TensorFlow; familiarity with agentic AI concepts and RL is essential. #J-18808-Ljbffr ...

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