Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction - View it on GitHub