Alternatives and similar repositories for dion

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

• modula-systems / modula
  🧱 Modula software package
  ☆321Updated 5 months ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆333Updated 2 months ago
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆306Updated last month
• huggingface / kernels
  Load compute kernels from the Hub
  ☆381Updated this week
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆197Updated last year
• meta-pytorch / torchforge
  PyTorch-native post-training at scale
  ☆600Updated this week
• google-deepmind / nanodo
  ☆289Updated last year
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆280Updated last month
• Noumena-Network / nmoe
  MoE training for Me and You and maybe other people
  ☆327Updated 3 weeks ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆185Updated last week
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆98Updated 6 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆472Updated 2 weeks ago
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆299Updated 5 months ago
• IST-DASLab / llmq
  Quantized LLM training in pure CUDA/C++.
  ☆233Updated last week
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆233Updated 2 weeks ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated last year
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆731Updated this week
• thinking-machines-lab / tinker
  Training API and CLI
  ☆318Updated last week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆305Updated last year
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆229Updated 7 months ago
• NVIDIA-NeMo / Emerging-Optimizers
  ☆127Updated last week
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆170Updated 6 months ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆363Updated last week
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆157Updated 2 months ago
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆113Updated 4 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆194Updated 3 weeks ago
• huggingface / picotron_tutorial
  ☆229Updated 2 months ago
• apple / ml-ademamix
  ☆70Updated last year
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆198Updated last year
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆818Updated last week