Alternatives and similar repositories for ml-cross-entropy

Users that are interested in ml-cross-entropy are comparing it to the libraries listed below

• KellerJordan / Muon
  Muon optimizer: +>30% sample efficiency with <3% wallclock overhead
  ☆623Updated last month
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆512Updated 6 months ago
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆233Updated 3 months ago
• hao-ai-lab / Consistency_LLM
  [ICML 2024] CLLMs: Consistency Large Language Models
  ☆391Updated 6 months ago
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆766Updated last week
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆245Updated this week
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆373Updated 2 weeks ago
• mit-han-lab / duo-attention
  [ICLR 2025] DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads
  ☆459Updated 3 months ago
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆327Updated 5 months ago
• zhuzilin / ring-flash-attention
  Ring attention implementation with flash attention
  ☆764Updated last month
• fla-org / native-sparse-attention
  🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"
  ☆665Updated last month
• haoliuhl / ringattention
  Large Context Attention
  ☆710Updated 3 months ago
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆481Updated last week
• NVlabs / hymba
  ☆177Updated 5 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆128Updated last week
• facebookresearch / LayerSkip
  Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024
  ☆294Updated 2 weeks ago
• sail-sg / oat
  🌾 OAT: A research-friendly framework for LLM online alignment, including preference learning, reinforcement learning, etc.
  ☆352Updated last week
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆214Updated 5 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆536Updated this week
• huggingface / picotron_tutorial
  ☆186Updated 3 months ago
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆412Updated this week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆324Updated last week
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆232Updated 2 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 9 months ago
• NVlabs / Minitron
  A family of compressed models obtained via pruning and knowledge distillation
  ☆336Updated 6 months ago
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆213Updated last year
• wolfecameron / nanoMoE
  An extension of the nanoGPT repository for training small MOE models.
  ☆142Updated 2 months ago
• apple / ml-sigmoid-attention
  ☆282Updated 3 weeks ago
• jzhang38 / EasyContext
  Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.
  ☆723Updated 7 months ago
• McGill-NLP / nano-aha-moment
  Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"
  ☆450Updated this week