rl4co
A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
how_attentive_are_gats
Code for the paper "How Attentive are Graph Attention Networks?" (ICLR'2022)
rlor
Reinforcement learning for operation research problems with OpenAI Gym and CleanRL
proteinn-structure-predictor
A transformer network trained to predict end-to-end single sequence protein structure as a set of angles given amino acid sequences.
lite_llama
A light llama-like llm inference framework based on the triton kernel.
https://github.com/google-research/scenic
Scenic: A Jax Library for Computer Vision Research and Beyond
gcvit
Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer
https://github.com/astrazeneca/selfpad
The official implementation of "Improving Antibody Humanness Prediction using Patent Data".
DeepMove
[WWW 2018] DeepMove: Predicting Human Mobility with Attentional Recurrent Network
fluence
A deep learning library based on Pytorch focussed on low resource language research and robustness
https://github.com/ai-forever/ner-bert
BERT-NER (nert-bert) with google bert https://github.com/google-research.
layer_norm_expressivity_role
Code for the paper "On the Expressivity Role of LayerNorm in Transformers' Attention" (Findings of ACL'2023)
https://github.com/christophreich1996/swin-transformer-v2
PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" [CVPR 2022].
multiscale-adversarial-attention-gates
Code for the paper: Valvano G. et al. (2021), Learning to Segment from Scribbles using Multi-scale Adversarial Attention Gates
cell-detr
Official and maintained implementation of the paper "Attention-Based Transformers for Instance Segmentation of Cells in Microstructures" [BIBM 2020].
https://github.com/adicksonlab/agdiff
Implementation of AGDIFF: Attention-Enhanced Diffusion for Molecular Geometry Prediction
reactive-mise-en-scene
Unity plugin using on-screen object position as proxy for player attention. Objects are assigned a "tendency" which, together with attention measurements, enables custom loading of prefabs to adapt narrative environments over time.
https://github.com/christophreich1996/maxvit
PyTorch reimplementation of the paper "MaxViT: Multi-Axis Vision Transformer" [ECCV 2022].
https://github.com/contextlab/attention-memory-task
An experiment used to explore interactions between covert attention and recognition memory
saits
The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model for efficient time-series imputation (impute multivariate incomplete time series containing NaN missing data/values with machine learning). https://arxiv.org/abs/2202.08516