Alternatives and similar repositories for neural-collapse

Users that are interested in neural-collapse are comparing it to the libraries listed below

• PrincetonUniversity / multi_gpu_training
  ☆367Updated 3 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆492Updated last year
• MinghuiChen43 / awesome-deep-phenomena
  A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...
  ☆382Updated last week
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆335Updated last month
• OFSkean / information_flow
  ☆27Updated 5 months ago
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆584Updated last year
• HazyResearch / flash-fft-conv
  FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
  ☆337Updated 11 months ago
• rbalestr-lab / stable-pretraining
  Reliable, minimal and scalable library for pretraining foundation and world models
  ☆110Updated last month
• AvivBick / awesome-ssm-ml
  Reading list for research topics in state-space models
  ☆338Updated 6 months ago
• google-research / jaxpruner
  ☆234Updated 10 months ago
• linusericsson / einspace
  Official code for our NeurIPS 2024 paper "einspace: Searching for Neural Architectures from Fundamental Operations"
  ☆30Updated last month
• mpezeshki / Epoch_wise_Double_Descent
  Official implementation of "Multi-scale Feature Learning Dynamics: Insights for Double Descent".
  ☆16Updated 3 years ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆193Updated last year
• LOG-postech / rethinking-LLM-pruning
  ☆28Updated 9 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆397Updated 3 months ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆404Updated last week
• HazyResearch / fly
  ☆221Updated 2 years ago
• insuhan / hyper-attn
  ☆83Updated 2 years ago
• Prisma-Multimodal / ViT-Prisma
  ViT Prisma is a mechanistic interpretability library for Vision and Video Transformers (ViTs).
  ☆327Updated 4 months ago
• mengxiayu / LLMSuperWeight
  Code for studying the super weight in LLM
  ☆121Updated last year
• rhubarbwu / linguistic-collapse
  Codebase for Linguistic Collapse: Neural Collapse in (Large) Language Models [NeurIPS 2024] [arXiv:2405.17767]
  ☆17Updated 8 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆585Updated 4 months ago
• kwignb / NeuralTangentKernel-Papers
  Neural Tangent Kernel Papers
  ☆120Updated 11 months ago
• topal-team / rockmate
  ☆36Updated last year
• daniel-geon-park / triton_bwd
  Automatic differentiation for Triton Kernels
  ☆29Updated 4 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆180Updated 5 months ago
• meta-pytorch / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆1,089Updated this week
• ModelZoos / ModelZooDataset
  Model Zoos published at the NeurIPS 2022 Dataset & Benchmark track: "Model Zoos: A Dataset of Diverse Populations of Neural Network Model…
  ☆57Updated 2 months ago
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆218Updated last year
• deep-spin / adasplash
  AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)
  ☆30Updated 2 months ago