shawntan / SUT

Related projects ⓘ

Alternatives and complementary repositories for SUT

• abhishekpanigrahi1996 / transformer_in_transformer
  ☆44Updated last year
• RobertCsordas / moe
  Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"
  ☆36Updated last year
• lucidrains / pause-transformer
  Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…
  ☆49Updated last year
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆53Updated 11 months ago
• Doraemonzzz / hgru2-pytorch
  ☆21Updated 2 months ago
• google-deepmind / spectral_ssm
  ☆29Updated 7 months ago
• berlino / seq_icl
  ☆50Updated 6 months ago
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆23Updated 5 months ago
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• sustcsonglin / mamba-triton
  ☆45Updated 9 months ago
• radarFudan / mamba-minimal-jax
  ☆29Updated this week
• sustcsonglin / gated_linear_attention_layer
  ☆31Updated 10 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 2 weeks ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated this week
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 10 months ago
• Doraemonzzz / nanoTransNormer
  ☆11Updated last year
• sjelassi / transformers_ssm_copy
  ☆24Updated 8 months ago
• IdoAmos / not-from-scratch
  ☆26Updated last month
• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆49Updated 3 months ago
• AndPotap / einsum-search
  ☆25Updated last month
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆31Updated last year
• Ryu1845 / hyena-jax
  Implementation of Hyena Hierarchy in JAX
  ☆10Updated last year
• bergen / EdgeTransformer
  ☆21Updated 2 years ago
• RobertCsordas / moe_layer
  sigma-MoE layer
  ☆18Updated 10 months ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆19Updated last year
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated 11 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆45Updated 3 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago