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
  ☆282Updated last month
• google-deepmind / nanodo
  ☆282Updated last year
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆296Updated last month
• meta-pytorch / monarch
  PyTorch Single Controller
  ☆435Updated this week
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆164Updated 3 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆83Updated 10 months ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆293Updated last week
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆265Updated last week
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆290Updated 2 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆95Updated 2 months ago
• huggingface / picotron_tutorial
  ☆222Updated last week
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆305Updated 2 months ago
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆190Updated 10 months ago
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆192Updated 10 months ago
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆181Updated 2 weeks ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆63Updated last week
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆650Updated this week
• cloneofsimo / min-fsdp
  ☆91Updated last year
• AI-Hypercomputer / maxdiffusion
  ☆264Updated this week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆292Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆167Updated 2 months ago
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆466Updated last week
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆247Updated 8 months ago
• divyamakkar0 / JAXformer
  A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
  ☆95Updated 2 weeks ago
• PrimeIntellect-ai / pi-quant
  SIMD quantization kernels
  ☆87Updated last month
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆194Updated 6 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆173Updated last year
• zaydzuhri / softpick-attention
  Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"
  ☆85Updated last month
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆189Updated 3 months ago
• thinking-machines-lab / batch_invariant_ops
  ☆811Updated last week