jysohn1108 / Looped-Transformer

Related projects ⓘ

Alternatives and complementary repositories for Looped-Transformer

• sustcsonglin / mamba-triton
  ☆45Updated 9 months ago
• abhishekpanigrahi1996 / transformer_in_transformer
  ☆44Updated last year
• berlino / seq_icl
  ☆50Updated 6 months ago
• Leiay / looped_transformer
  ☆20Updated 11 months ago
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆24Updated 7 months ago
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆34Updated last week
• McGill-NLP / VinePPO
  Code for the paper "VinePPO: Unlocking RL Potential For LLM Reasoning Through Refined Credit Assignment"
  ☆80Updated last week
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆27Updated 5 months ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆79Updated last year
• lee-ny / teaching_arithmetic
  ☆70Updated last year
• RobertCsordas / moeut
  ☆62Updated 3 months ago
• HazyResearch / prefix-linear-attention
  ☆45Updated 4 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆45Updated 3 weeks ago
• r-three / RAD
  Reference implementation for Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model
  ☆41Updated 10 months ago
• sustcsonglin / gated_linear_attention_layer
  ☆31Updated 10 months ago
• shunzh / Code-AI-Tree-Search
  ☆105Updated 4 months ago
• gregorbachmann / Next-Token-Failures
  ☆69Updated 8 months ago
• maximzubkov / fft-scan
  Efficient PScan implementation in PyTorch
  ☆15Updated 10 months ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆40Updated 3 months ago
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆23Updated 5 months ago
• wellecks / lm-evaluation-harness
  A framework for few-shot evaluation of autoregressive language models.
  ☆23Updated 11 months ago
• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆49Updated 3 months ago
• abaheti95 / LoL-RL
  Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients
  ☆26Updated 2 months ago
• allenai / easy-to-hard-generalization
  Code for the arXiv preprint "The Unreasonable Effectiveness of Easy Training Data"
  ☆44Updated 10 months ago
• Yu-Fangxu / FoR
  Flow of Reasoning: Training LLMs for Divergent Problem Solving with Minimal Examples
  ☆39Updated last month
• IdoAmos / not-from-scratch
  ☆25Updated last month
• yikangshen / megablocks
  ☆18Updated 5 months ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆61Updated 7 months ago
• princeton-nlp / TransformerPrograms
  [NeurIPS 2023] Learning Transformer Programs
  ☆157Updated 6 months ago
• kamanphoebe / Look-into-MoEs
  A Closer Look into Mixture-of-Experts in Large Language Models
  ☆40Updated 3 months ago