msa-transformer

starcoder

Home of StarCoder: fine-tuning & inference!

kaggle_sartorius

Solution for the Sartorius - Cell Instance Segmentation competition

Attention-OCR

Visual Attention based OCR

minGPT-flax

GPT implementation in Flax

flash-attention

Fast and memory-efficient exact attention

sentiment-discovery

Unsupervised Language Modeling at scale for robust sentiment classification

RoseTTAFold

This package contains deep learning models and related scripts for RoseTTAFold

bilm-tf

Tensorflow implementation of contextualized word representations from bi-directional language models

grouped-query-attention-pytorch

(Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generali...

performer-pytorch

An implementation of Performer, a linear attention-based transformer, in Pytorch

se3-transformer-pytorch

Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repo...

favor

Simple implementation of FAVOR attention layer

molecule-attention-transformer

Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle th...

torchinfo

View model summaries in PyTorch!

mistral-finetune