Snabbfakta

    • London

Ansök senast: 2025-02-20

ML Research Engineer

Publicerad 2024-12-22

Job Description

About the role:

The job opening is part of a research project funded by the ARIA program: “Scaling Compute” by bringing the cost of AI hardware down by >1000x. The project is about Equilibrium Propagation (EP), an alternative training framework to backpropagation (BP) that is compatible with analog computing hardware (i.e. fast and energy-efficient hardware). Specifically, the project is aimed at demonstrating through simulations that EP can be a viable alternative to BP to solve modern ML tasks on analog computing platforms.


In this position, you will help develop a software framework for EP in PyTorch. This framework, which will support both hardware and software simulations, will enable scaling of EP to large networks and datasets, enabling the core experiments of the research project.


Responsibilities:

  • Developing a software framework for the simulations of EP (in PyTorch), building upon the one available at this link
  • Developing unit tests and establishing a working pipeline for us to safely contribute to the framework as we scale it
  • Making the framework parallelizable on multiple GPUs (parallelization across mini-batches of data, parallelization over the computation of different equilibrium states of EP, etc.)
  • Developing tools to store experimental results in an organized way, analyze and visualize the data/re...