bytedance / effective_transformer
Running BERT without Padding
☆471Updated 3 years ago
Alternatives and similar repositories for effective_transformer:
Users that are interested in effective_transformer are comparing it to the libraries listed below
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆473Updated last year
- Transformer related optimization, including BERT, GPT☆59Updated last year
- Zero Bubble Pipeline Parallelism☆386Updated 3 weeks ago
- ☆215Updated last year
- ☆411Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆266Updated 11 months ago
- Microsoft Automatic Mixed Precision Library☆596Updated 7 months ago
- Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.☆267Updated 2 years ago
- Latency and Memory Analysis of Transformer Models for Training and Inference☆407Updated 2 weeks ago
- ☆117Updated last year
- ☆139Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆120Updated 4 months ago
- Simple Dynamic Batching Inference☆145Updated 3 years ago
- DeepLearning Framework Performance Profiling Toolkit☆284Updated 3 years ago
- Best practice for training LLaMA models in Megatron-LM☆650Updated last year
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆144Updated this week
- ☆127Updated 4 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Disaggregated serving system for Large Language Models (LLMs).☆575Updated 3 weeks ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆488Updated 2 weeks ago
- A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster☆157Updated last year
- Pipeline Parallelism for PyTorch☆765Updated 8 months ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆403Updated this week
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆405Updated last week
- This repository contains integer operators on GPUs for PyTorch.☆203Updated last year
- Fast implementation of BERT inference directly on NVIDIA (CUDA, CUBLAS) and Intel MKL☆545Updated 4 years ago
- Implementation of a Transformer, but completely in Triton☆264Updated 3 years ago
- ☆314Updated last year
- a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.☆1,520Updated 3 weeks ago