Alternatives and similar repositories for RTF:

Users that are interested in RTF are comparing it to the libraries listed below

• shikaiqiu / compute-better-spent
  ☆52Updated 5 months ago
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆122Updated last year
• nikhilvyas / SOAP
  ☆161Updated 3 months ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆87Updated 8 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• apple / ml-ademamix
  ☆57Updated 3 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 3 months ago
• epfml / DenseFormer
  ☆78Updated 10 months ago
• google-deepmind / spectral_ssm
  ☆31Updated 10 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆96Updated 2 months ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆92Updated last week
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆63Updated 2 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆85Updated 6 months ago
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆61Updated this week
• jopetty / word-problem
  Experiments on the impact of depth in transformers and SSMs.
  ☆23Updated 3 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆93Updated 6 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆51Updated last month
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆54Updated 4 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆137Updated last week
• TariqAHassan / S4Torch
  PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.
  ☆77Updated last year
• proger / hippogriff
  Griffin MQA + Hawk Linear RNN Hybrid
  ☆85Updated 10 months ago
• AndPotap / einsum-search
  ☆30Updated 4 months ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆105Updated 2 months ago
• sustcsonglin / flash-linear-rnn
  Implementations of various linear RNN layers using pytorch and triton
  ☆49Updated last year
• lindermanlab / S5
  ☆283Updated last month
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆23Updated last month
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆75Updated 4 months ago