huggingface / picotron_tutorial
β181Updated 2 months ago
Alternatives and similar repositories for picotron_tutorial:
Users that are interested in picotron_tutorial are comparing it to the libraries listed below
- π Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flashβ¦β244Updated this week
- Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, sparsβ¦β322Updated 4 months ago
- An extension of the nanoGPT repository for training small MOE models.β138Updated last month
- A repository to unravel the language of GPUs, making their kernel conversations easy to understandβ180Updated last week
- Best practices & guides on how to write distributed pytorch training codeβ406Updated 2 months ago
- PyTorch building blocks for the OLMo ecosystemβ205Updated this week
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024β291Updated this week
- ring-attention experimentsβ132Updated 6 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β105Updated this week
- LLM KV cache compression made easyβ471Updated this week
- πΎ OAT: A research-friendly framework for LLM online alignment, including preference learning, reinforcement learning, etc.β336Updated 2 weeks ago
- Scalable and Performant Data Loadingβ247Updated this week
- β155Updated last year
- A project to improve skills of large language modelsβ354Updated this week
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)β71Updated this week
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"β231Updated 3 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.β536Updated this week
- Efficient triton implementation of Native Sparse Attention.β142Updated 3 weeks ago
- Triton-based implementation of Sparse Mixture of Experts.β212Updated 5 months ago
- Normalized Transformer (nGPT)β174Updated 5 months ago
- β186Updated this week
- β78Updated 10 months ago
- Load compute kernels from the Hubβ115Updated last week
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top ofβ¦β130Updated 8 months ago
- prime-rl is a codebase for decentralized RL training at scaleβ85Updated this week
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clustersβ126Updated 5 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"β232Updated 2 months ago
- A scalable asynchronous reinforcement learning implementation with in-flight weight updates.β94Updated this week
- minimal GRPO implementation from scratchβ87Updated last month
- Explorations into some recent techniques surrounding speculative decodingβ261Updated 4 months ago