Alternatives and similar repositories for grow-crystals

Users that are interested in grow-crystals are comparing it to the libraries listed below

• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆101Updated 8 months ago
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆190Updated 8 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆96Updated last month
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆184Updated 11 months ago
• epfml / DenseFormer
  ☆82Updated last year
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆67Updated last year
• microsoft / dion
  Dion optimizer algorithm
  ☆305Updated this week
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆101Updated last month
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆283Updated last month
• apoorvkh / academic-pretraining
  $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources
  ☆143Updated 3 months ago
• nikhilvyas / SOAP
  ☆207Updated 8 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆254Updated 3 weeks ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆289Updated 2 months ago
• JoeLi12345 / nGPT
  an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)
  ☆105Updated 5 months ago
• PolymathicAI / xVal
  Repository for code used in the xVal paper
  ☆141Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆79Updated last year
• ShadeAlsha / ICon
  ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"
  ☆111Updated 2 months ago
• apple / ml-sigma-reparam
  ☆307Updated last year
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆131Updated 2 weeks ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆76Updated 8 months ago
• BlackHC / neural_net_checklist
  ☆150Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated 8 months ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated 9 months ago
• NX-AI / xlstm-jax
  Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…
  ☆101Updated 7 months ago
• Think-a-Tron / evolve
  open source alpha evolve
  ☆67Updated 3 months ago
• zaydzuhri / softpick-attention
  Implementations of attention with the softpick function, naive and FlashAttention-2
  ☆83Updated 3 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆184Updated 7 months ago
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆191Updated last year
• apple / ml-ademamix
  ☆65Updated 9 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆225Updated last week