Shawn Tan's repositories
scattermoe
Triton-based implementation of Sparse Mixture of Experts.
theano_toolkit
Collection of useful, re-used routines.
icentia-ecg
Working on Icentia ECG data.
chicken-rice-nn
Miscellaneous code for doing NLP with Theano
theano-kaldi
Bunch of scripts for working with Kaldi.
stack-binary-recursive-nn
Parallelised implementation of Recursive Neural Networks for binary trees in PyTorch
lost-in-the-middle
Code and data for "Lost in the Middle: How Language Models Use Long Contexts"
stick-breaking-vae
Infinite MIxture Models with VAEs
awd-lstm-lm
LSTM and QRNN Language Model Toolkit for PyTorch
CategoricalNF
Official repository for "Categorical Normalizing Flows via Continuous Transformations"
lm-evaluation-harness
A framework for few-shot evaluation of autoregressive language models.
transformer_latent_diffusion
Text to Image Latent Diffusion using a Transformer core
transformers
🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
zoology
Understand and test language model architectures on synthetic tasks.