rasbt / pytorch-memory-optim

Related projects ⓘ

Alternatives and complementary repositories for pytorch-memory-optim

• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆67Updated this week
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆119Updated 3 months ago
• RobertCsordas / moe_attention
  Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"
  ☆92Updated last month
• cloneofsimo / min-fsdp
  ☆73Updated 4 months ago
• rasbt / dora-from-scratch
  LoRA and DoRA from Scratch Implementations
  ☆188Updated 8 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆133Updated 9 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆57Updated 11 months ago
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆157Updated 10 months ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆97Updated last month
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆84Updated last week
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆66Updated 5 months ago
• SeunghyunSEO / optimized_hf_llama_class_for_training
  ☆45Updated 2 months ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆80Updated 11 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆113Updated 7 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated 11 months ago
• Pleias / Various-Finetuning
  Set of scripts to finetune LLMs
  ☆36Updated 7 months ago
• euclaise / supertrainer2000
  ☆49Updated 8 months ago
• epfml / DenseFormer
  ☆76Updated 7 months ago
• hahnyuan / PB-LLM
  PB-LLM: Partially Binarized Large Language Models
  ☆148Updated last year
• itsnamgyu / block-transformer
  Block Transformer: Global-to-Local Language Modeling for Fast Inference (Official Code)
  ☆135Updated last month
• tcapelle / llm_recipes
  A set of scripts and notebooks on LLM finetunning and dataset creation
  ☆93Updated last month
• rasbt / cvpr2023
  ☆127Updated last year
• melisa-writer / short-transformers
  Prune transformer layers
  ☆64Updated 5 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆104Updated last month
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆59Updated 3 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆82Updated 8 months ago
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆104Updated last month
• minyoungg / LTE
  ☆63Updated 4 months ago
• eth-easl / fmengine
  Utilities for Training Very Large Models
  ☆56Updated last month
• IST-DASLab / SparseFinetuning
  Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry
  ☆38Updated 10 months ago