awslabs / lorien
☆44Updated last year
Related projects ⓘ
Alternatives and complementary repositories for lorien
- ☆23Updated 10 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆114Updated 2 years ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- ☆140Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆131Updated last year
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆49Updated 3 months ago
- Benchmark PyTorch Custom Operators☆13Updated last year
- DietCode Code Release☆62Updated 2 years ago
- A schedule language for large model training☆141Updated 5 months ago
- ☆90Updated 2 years ago
- Repository for SysML19 Artifacts Evaluation☆53Updated 5 years ago
- An IR for efficiently simulating distributed ML computation.☆25Updated 10 months ago
- Slides from 2021-12-15 talk, "TVM Developer Bootcamp – Writing Hardware Backends"☆10Updated 2 years ago
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆17Updated 2 years ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆34Updated 2 years ago
- Benchmark scripts for TVM☆73Updated 2 years ago
- Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation☆26Updated 5 years ago
- Boost hardware utilization for ML training workloads via Inter-model Horizontal Fusion☆32Updated 6 months ago
- FTPipe and related pipeline model parallelism research.☆41Updated last year
- ☆65Updated 3 years ago
- ☆73Updated last year
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆63Updated 2 years ago
- Cavs: An Efficient Runtime System for Dynamic Neural Networks☆13Updated 4 years ago
- ☆47Updated last year
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆29Updated 3 months ago
- Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into ef…☆60Updated 2 years ago
- ☆8Updated last year
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆24Updated 4 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆75Updated this week
- A library for syntactically rewriting Python programs, pronounced (sinner).☆70Updated 2 years ago