5 of 5 JAX Jobs in the East of England

Solid grounding in machine learning/deep-learning fundamentals — you can build, train, and debug neural networks from scratch in a modern framework (PyTorch, JAX, or similar). Demonstrable hands-on experience: research projects, publications, internships, open-source, or production work. Comfortable reading ML papers and reimplementing them. A fast ...

Automation Skills Essential: Competent in writing Shell Scripts. Desirable: Experience of C, Python or Fortran programming. Experience of installing and using PyTORCH, Tensorflow, openCV & Jax Understanding the work of the Laboratory Essential: Experience working in an IT support, academic or research environment. Desirable: Research experience in Biology. People Skills, communication ...

Nice to have Experience with optimisation, reinforcement learning and/or large language models Familiarity with deep learning frameworks such as TensorFlow, PyTorch or JAX Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research A track record of contributing ...

experience in academia or industry. We will consider various levels of experience in relevant research areas. Key Skills Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile ...

Succeed Experience building, training, adapting, or evaluating machine learning models. Strong software engineering skills in Python and modern ML frameworks such as PyTorch, JAX, or TensorFlow. Experience designing experiments, evaluation metrics, or test sets for model performance. Ability to debug model behavior using data, traces, logs, and qualitative feedback. Experience ...