Tong Zheng's repositories
PartialFormer
Official code of PartialFormer (ACL2024 Findings)
attention-is-all-you-need-pytorch
A PyTorch implementation of the Transformer model in "Attention is All You Need".
HCP_download
Script to download data from the Human Connectome Project, HCP-1200 subjects release.
EIT-Enhanced-Interactive-Transformer
Official code of EIT: Enhanced Interactive Transformer (ACL2024 Main)
fast_robust_early_exit
Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)
fmridenoise
Tool for automatic denoising, denoising strategies comparison, and functional connectivity data quality control.
GLAT
Implementation of "Glancing Transformer for Non-Autoregressive Neural Machine Translation"
GNNPapers
Must-read papers on graph neural networks (GNN)
Introduction-to-Transformers
An introduction to Transformers
lex-glue
LexGLUE: A Benchmark Dataset for Legal Language Understanding in English
llama
Inference code for LLaMA models
PRGNN_fMRI
Pytorch implementation of pooling-regularized GNN (PRGNN) for fMRI analysis. https://arxiv.org/pdf/2007.14589.pdf
PYTORCH_MTCNN
MTCNN BASE PYTORCH
torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.