5 of 5 JAX Jobs in the South East

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

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