Ascend / pytorch
Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch
☆294Updated this week
Alternatives and similar repositories for pytorch:
Users that are interested in pytorch are comparing it to the libraries listed below
- FlagScale is a large model toolkit based on open-sourced projects.☆209Updated this week
- FlagGems is an operator library for large language models implemented in Triton Language.☆406Updated this week
- Disaggregated serving system for Large Language Models (LLMs).☆453Updated 5 months ago
- RTP-LLM: Alibaba's high-performance LLM inference engine for diverse applications.☆607Updated last week
- DashInfer is a native LLM inference engine aiming to deliver industry-leading performance atop various hardware architectures, including …☆220Updated this week
- [EMNLP 2024 Industry Track] This is the official PyTorch implementation of "LLMC: Benchmarking Large Language Model Quantization with a V…☆390Updated last week
- A collection of memory efficient attention operators implemented in the Triton language.☆233Updated 7 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆415Updated 3 weeks ago
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆219Updated 2 weeks ago
- ☆127Updated last month
- ☆140Updated 9 months ago
- Analyze the inference of Large Language Models (LLMs). Analyze aspects like computation, storage, transmission, and hardware roofline mod…☆377Updated 4 months ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆469Updated 10 months ago
- Best practice for training LLaMA models in Megatron-LM☆641Updated last year
- ☆151Updated last month
- Zero Bubble Pipeline Parallelism☆317Updated 2 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆273Updated last month
- A fast communication-overlapping library for tensor parallelism on GPUs.☆280Updated 3 months ago
- GLake: optimizing GPU memory management and IO transmission.☆424Updated 2 months ago
- ☆76Updated last year
- ☆311Updated last week
- A PyTorch Native LLM Training Framework☆698Updated last month
- QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving☆492Updated this week
- export llama to onnx☆112Updated last month
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆338Updated last week
- The official repo of Pai-Megatron-Patch for LLM & VLM large scale training developed by Alibaba Cloud.☆807Updated this week
- A throughput-oriented high-performance serving framework for LLMs☆714Updated 4 months ago
- ☆142Updated 2 weeks ago
- Ring attention implementation with flash attention☆660Updated last month
- A baseline repository of Auto-Parallelism in Training Neural Networks☆142Updated 2 years ago