awslabs / raf
☆141Updated last year
Related projects: ⓘ
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆191Updated 2 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆112Updated 2 years ago
- ☆86Updated last year
- ☆193Updated last year
- A home for the final text of all TVM RFCs.☆99Updated 3 months ago
- DietCode Code Release☆59Updated 2 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆129Updated last year
- ☆23Updated 7 months ago
- ☆45Updated last year
- Shared Middle-Layer for Triton Compilation☆157Updated last week
- A schedule language for large model training☆139Updated 3 months ago
- ☆138Updated 2 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆49Updated last month
- Benchmark scripts for TVM☆73Updated 2 years ago
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆175Updated 2 years ago
- ☆72Updated last year
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆69Updated 3 years ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆184Updated this week
- ☆34Updated 2 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆70Updated last month
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆129Updated last year
- ☆127Updated last month
- A baseline repository of Auto-Parallelism in Training Neural Networks☆138Updated 2 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆265Updated this week
- Experimental projects related to TensorRT☆62Updated this week
- play gemm with tvm☆81Updated last year
- ☆62Updated 3 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆127Updated 2 years ago
- Assembler for NVIDIA Volta and Turing GPUs☆195Updated 2 years ago