Ascend / pytorchLinks
Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch
☆483Updated this week
Alternatives and similar repositories for pytorch
Users that are interested in pytorch are comparing it to the libraries listed below
Sorting:
- FlagScale is a large model toolkit based on open-sourced projects.☆471Updated last week
- DashInfer is a native LLM inference engine aiming to deliver industry-leading performance atop various hardware architectures, including …☆274Updated 6 months ago
- RTP-LLM: Alibaba's high-performance LLM inference engine for diverse applications.☆1,037Updated last week
- ☆523Updated 2 weeks ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆476Updated last year
- FlagGems is an operator library for large language models implemented in the Triton Language.☆893Updated this week
- [EMNLP 2024 & AAAI 2026] A powerful toolkit for compressing large models including LLMs, VLMs, and video generative models.☆672Updated 2 months ago
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆298Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆287Updated last year
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆676Updated this week
- ☆130Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆641Updated 3 weeks ago
- Analyze the inference of Large Language Models (LLMs). Analyze aspects like computation, storage, transmission, and hardware roofline mod…☆615Updated last year
- ☆73Updated last year
- ☆141Updated last year
- GLake: optimizing GPU memory management and IO transmission.☆497Updated 10 months ago
- ☆437Updated 4 months ago
- ☆74Updated this week
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆417Updated 5 months ago
- ☆155Updated 11 months ago
- ☆76Updated last year
- Zero Bubble Pipeline Parallelism☆449Updated 9 months ago
- Disaggregated serving system for Large Language Models (LLMs).☆771Updated 10 months ago
- Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs☆923Updated 2 months ago
- Ascend TileLang adapter☆206Updated this week
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆810Updated 11 months ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆248Updated 2 weeks ago
- Transformer related optimization, including BERT, GPT☆59Updated 2 years ago
- Materials for learning SGLang☆738Updated last month
- Fast and memory-efficient exact attention☆111Updated last week