2 of 2 JAX Jobs in East Anglia

Deep Learning Researcher

Hiring Organisation
Microtech Global Ltd
Location
Cambridge, Cambridgeshire, East Anglia, United Kingdom
Employment Type
Permanent
Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research ...

ML Research Scientist

Hiring Organisation
Microtech Global Ltd
Location
Cambridge, Cambridgeshire, East Anglia, United Kingdom
Employment Type
Permanent
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 ...