search

Genentech / gReLU

gReLU is a python library to train, interpret, and apply deep learning models to DNA sequences.
https://genentech.github.io/gReLU/
MIT License
230 stars 23 forks source link

Support fast attention #64

Closed avantikalal closed 1 month ago

avantikalal commented 1 month ago

Add support for fast attention layers in https://github.com/Genentech/gReLU/blob/main/src/grelu/model/layers.py.

suragnair commented 1 month ago

Should we mark this as complete?