Ascend / pytorch
Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch
☆305Updated this week
Alternatives and similar repositories for pytorch:
Users that are interested in pytorch are comparing it to the libraries listed below
- DashInfer is a native LLM inference engine aiming to deliver industry-leading performance atop various hardware architectures, including …☆236Updated 2 weeks ago
- Analyze the inference of Large Language Models (LLMs). Analyze aspects like computation, storage, transmission, and hardware roofline mod…☆406Updated 5 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆243Updated this week
- FlagGems is an operator library for large language models implemented in Triton Language.☆429Updated this week
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆440Updated last week
- A collection of memory efficient attention operators implemented in the Triton language.☆242Updated 8 months ago
- Disaggregated serving system for Large Language Models (LLMs).☆472Updated 6 months ago
- ☆139Updated 10 months ago
- ☆127Updated 2 months ago
- ☆317Updated last month
- [EMNLP 2024 Industry Track] This is the official PyTorch implementation of "LLMC: Benchmarking Large Language Model Quantization with a V…☆421Updated this week
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆469Updated 11 months ago
- GLake: optimizing GPU memory management and IO transmission.☆433Updated 3 months ago
- ☆116Updated this week
- RTP-LLM: Alibaba's high-performance LLM inference engine for diverse applications.☆641Updated last month
- Best practice for training LLaMA models in Megatron-LM☆646Updated last year
- ☆154Updated this week
- Zero Bubble Pipeline Parallelism☆345Updated 3 weeks ago
- ☆40Updated this week
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆223Updated 3 weeks ago
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆427Updated 7 months ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆742Updated 6 months ago
- Ring attention implementation with flash attention☆692Updated last week
- PyTorch bindings for CUTLASS grouped GEMM.☆98Updated 2 months ago
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆559Updated last week
- ☆142Updated last month
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆92Updated 11 months ago
- Materials for learning SGLang☆299Updated this week
- A fast communication-overlapping library for tensor parallelism on GPUs.☆319Updated this week