Published July 10, 2025 | Version v1

GainCellAttention

Description

This repository contains the codes used to train the models discussed in the article :

N. Leroux, P.-P. Manea, C. Sudarshan, J. Finkbeiner, S. Siegel, J. P. Strachan, and E. Neftci, Analog In-Memory Computing Attention Mechanism for Fast and Energy-Efficient Large Language Models, arXiv:2409.19315. https://arxiv.org/abs/2409.19315.

Files

GainCellAttention-1.0(1).zip

Files (8.4 MB)

Name Size
md5:19b895f8df070d22905f485dc8f5a96e
8.4 MB Preview Download

Additional details

Software

Repository URL
https://github.com/NathanLeroux-git/GainCellAttention.git
Programming language
Python
Development Status
Active