toshas / torch-discounted-cumsum

Related projects: ⓘ

• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆78Updated 2 years ago
• ischlag / fast-weight-transformers
  Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.
  ☆97Updated 3 years ago
• RobertCsordas / transformer_generalization
  The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…
  ☆66Updated last year
• choidami / sst
  ☆49Updated 3 years ago
• lucidrains / HTM-pytorch
  Implementation of Hierarchical Transformer Memory (HTM) for Pytorch
  ☆73Updated 3 years ago
• lucidrains / compressive-transformer-pytorch
  Pytorch implementation of Compressive Transformers, from Deepmind
  ☆154Updated 2 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• harvardnlp / genbmm
  CUDA kernels for generalized matrix-multiplication in PyTorch
  ☆79Updated 2 years ago
• ssnl / PyTorch-Reparam-Module
  Reparameterize your PyTorch modules
  ☆70Updated 3 years ago
• davisyoshida / haiku-mup
  A port of muP to JAX/Haiku
  ☆25Updated last year
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆94Updated last year
• srush / torch-queue
  ☆64Updated 10 months ago
• cpcp1998 / PermuteFormer
  Code for the paper PermuteFormer
  ☆43Updated 2 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆46Updated 2 years ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆38Updated last year
• johnryan465 / pscan
  ☆33Updated 8 months ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆65Updated 11 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆183Updated 3 months ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆94Updated last year
• norabelrose / transformers-plus-performers
  ☆17Updated this week
• microsoft / ReinMax
  Beyond Straight-Through
  ☆82Updated last year
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 3 years ago
• Thrandis / EKFAC-pytorch
  Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
  ☆139Updated last year
• anirudh9119 / neural_production_systems
  ☆14Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆36Updated 5 months ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• danijar / elements
  Building blocks for productive research
  ☆44Updated last week
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆47Updated 2 years ago