state-spaces / mamba

Related projects ⓘ

Alternatives and complementary repositories for mamba

• johnma2006 / mamba-minimal
  Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
  ☆2,624Updated 8 months ago
• Dao-AILab / flash-attention
  Fast and memory-efficient exact attention
  ☆14,279Updated this week
• hustvl / Vim
  [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
  ☆2,993Updated last week
• facebookresearch / DiT
  Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"
  ☆6,369Updated 5 months ago
• microsoft / LoRA
  Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"
  ☆10,776Updated 3 months ago
• state-spaces / s4
  Structured state space sequence models
  ☆2,470Updated 4 months ago
• alxndrTL / mamba.py
  A simple and efficient Mamba implementation in pure PyTorch and MLX.
  ☆1,012Updated 2 months ago
• MzeroMiko / VMamba
  VMamba: Visual State Space Models，code is based on mamba
  ☆2,196Updated 3 weeks ago
• huggingface / peft
  🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
  ☆16,471Updated this week
• huggingface / accelerate
  🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…
  ☆7,958Updated this week
• openai / transformer-debugger
  ☆4,035Updated 5 months ago
• salesforce / LAVIS
  LAVIS - A One-stop Library for Language-Vision Intelligence
  ☆9,943Updated this week
• test-time-training / ttt-lm-pytorch
  Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States
  ☆1,040Updated 4 months ago
• KindXiaoming / pykan
  Kolmogorov Arnold Networks
  ☆15,083Updated this week
• facebookresearch / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆8,660Updated this week
• microsoft / unilm
  Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
  ☆20,194Updated last week
• yyyujintang / Awesome-Mamba-Papers
  Awesome Papers related to Mamba.
  ☆1,220Updated last month
• pytorch / torchtune
  PyTorch native finetuning library
  ☆4,336Updated this week
• bitsandbytes-foundation / bitsandbytes
  Accessible large language models via k-bit quantization for PyTorch.
  ☆6,299Updated this week
• lucidrains / denoising-diffusion-pytorch
  Implementation of Denoising Diffusion Probabilistic Model in Pytorch
  ☆8,373Updated last month
• google-research / vision_transformer
  ☆10,467Updated 6 months ago
• facebookresearch / sam2
  The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained mode…
  ☆12,439Updated last month
• huggingface / trl
  Train transformer language models with reinforcement learning.
  ☆10,086Updated this week
• BlinkDL / RWKV-LM
  RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best…
  ☆12,672Updated this week
• NVIDIA / Megatron-LM
  Ongoing research training transformer models at scale
  ☆10,595Updated this week
• haotian-liu / LLaVA
  [NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.
  ☆20,286Updated 3 months ago
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆5,890Updated 7 months ago
• QwenLM / Qwen-VL
  The official repo of Qwen-VL (通义千问-VL) chat & pretrained large vision language model proposed by Alibaba Cloud.
  ☆5,055Updated 3 months ago
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,034Updated 7 months ago
• pytorch-labs / gpt-fast
  Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
  ☆5,669Updated last month