Implement and optimize attention mechanisms in neu | COFYT