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
  ☆322Updated 5 months ago
• Noumena-Network / nmoe
  MoE training for Me and You and maybe other people
  ☆335Updated last month
• meta-pytorch / torchforge
  PyTorch-native post-training at scale
  ☆613Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆389Updated last week
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆307Updated 2 months ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆334Updated 3 months ago
• google-deepmind / nanodo
  ☆289Updated last year
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆198Updated last year
• NVIDIA-NeMo / Emerging-Optimizers
  ☆147Updated this week
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆281Updated last month
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆236Updated 3 weeks ago
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆115Updated 4 months ago
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆175Updated this week
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated last year
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆186Updated 2 weeks ago
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆752Updated this week
• IST-DASLab / llmq
  Quantized LLM training in pure CUDA/C++.
  ☆235Updated 2 weeks ago
• thinking-machines-lab / tinker
  Training API and CLI
  ☆325Updated last week
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆300Updated 6 months ago
• apple / ml-ademamix
  ☆70Updated last year
• thinking-machines-lab / batch_invariant_ops
  ☆957Updated 3 months ago
• huggingface / picotron_tutorial
  ☆232Updated 2 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆170Updated 6 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆352Updated 2 months ago
• cloneofsimo / min-fsdp
  ☆92Updated last year
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆157Updated 2 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆475Updated this week
• AI-Hypercomputer / maxdiffusion
  ☆304Updated last week
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆827Updated last week
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆364Updated this week