Alternatives and similar repositories for kron_torch

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

• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆82Updated 7 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆232Updated last week
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated 11 months ago
• epfml / DenseFormer
  ☆81Updated last year
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆191Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆204Updated 3 months ago
• nikhilvyas / SOAP
  ☆197Updated 7 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆183Updated 10 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆76Updated 6 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆101Updated 6 months ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆288Updated this week
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆97Updated this week
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆189Updated 7 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆128Updated 2 weeks ago
• cloneofsimo / min-fsdp
  ☆79Updated last year
• VatsaDev / NanoPoor
  NanoGPT-speedrunning for the poor T4 enjoyers
  ☆68Updated 2 months ago
• apoorvkh / academic-pretraining
  $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources
  ☆140Updated last month
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆129Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated 11 months ago
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆118Updated 4 months ago
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆98Updated 5 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆65Updated 11 months ago
• zaydzuhri / softpick-attention
  Implementations of attention with the softpick function, naive and FlashAttention-2
  ☆80Updated 2 months ago
• kvfrans / splus
  ☆110Updated last month
• apple / ml-ademamix
  ☆61Updated 8 months ago
• fal-ai-community / NativeSparseAttention
  research impl of Native Sparse Attention (2502.11089)
  ☆54Updated 4 months ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆92Updated last month
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• apple / ml-sigma-reparam
  ☆303Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆521Updated this week