BiEchi / DistributedTrainingGPT2Links
基于PyTorch GPT-2的针对各种数据并行pretrain的研究代码.
☆11Updated 3 years ago
Alternatives and similar repositories for DistributedTrainingGPT2
Users that are interested in DistributedTrainingGPT2 are comparing it to the libraries listed below
Sorting:
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆113Updated 10 months ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆71Updated 2 years ago
- Odysseus: Playground of LLM Sequence Parallelism☆79Updated last year
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆47Updated last year
- Repository of LV-Eval Benchmark☆73Updated last year
- Research without Re-search: Maximal Update Parametrization Yields Accurate Loss Prediction across Scales☆32Updated 2 years ago
- ☆15Updated 2 years ago
- ☆132Updated 8 months ago
- Low-bit optimizers for PyTorch☆138Updated 2 years ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆98Updated 2 years ago
- Multi-Candidate Speculative Decoding☆39Updated last year
- Codes for our paper "Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generation" (EMNLP 2023 Findings)☆46Updated 2 years ago
- ☆79Updated 2 years ago
- Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**☆214Updated 11 months ago
- ☆125Updated last year
- Triton implementation of Flash Attention2.0☆47Updated 2 years ago
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆152Updated 10 months ago
- Distributed IO-aware Attention algorithm☆23Updated 4 months ago
- qwen-nsa☆87Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆184Updated last month
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆65Updated last year
- ☆87Updated this week
- (best/better) practices of megatron on veRL and tuning guide☆127Updated 4 months ago
- A LLaMA1/LLaMA12 Megatron implement.☆28Updated 2 years ago
- ☆158Updated 11 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆287Updated last year
- ☆22Updated 2 years ago
- A prototype repo for hybrid training of pipeline parallel and distributed data parallel with comments on core code snippets. Feel free to…☆57Updated 2 years ago
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆73Updated 6 months ago