awslabs / lorien
☆44Updated last year
Alternatives and similar repositories for lorien:
Users that are interested in lorien are comparing it to the libraries listed below
- ☆23Updated last month
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆117Updated 2 years ago
- DietCode Code Release☆61Updated 2 years ago
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆18Updated 3 years ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- ☆141Updated last week
- Slides from 2021-12-15 talk, "TVM Developer Bootcamp – Writing Hardware Backends"☆10Updated 2 years ago
- Cavs: An Efficient Runtime System for Dynamic Neural Networks☆13Updated 4 years ago
- Benchmark PyTorch Custom Operators☆13Updated last year
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆73Updated 4 years ago
- FTPipe and related pipeline model parallelism research.☆41Updated last year
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated 5 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆133Updated last year
- ☆92Updated 2 years ago
- ☆72Updated 2 years ago
- Repository for SysML19 Artifacts Evaluation☆53Updated 5 years ago
- A schedule language for large model training☆143Updated 7 months ago
- An Attention Superoptimizer☆20Updated 8 months ago
- An external memory allocator example for PyTorch.☆14Updated 3 years ago
- Benchmark scripts for TVM☆73Updated 2 years ago
- 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
- This is the implementation for paper: AdaTune: Adaptive Tensor Program CompilationMade Efficient (NeurIPS 2020).☆13Updated 3 years ago
- ☆70Updated 3 years ago
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆25Updated 4 years ago
- GVProf: A Value Profiler for GPU-based Clusters☆48Updated 9 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…☆61Updated 2 years ago
- Boost hardware utilization for ML training workloads via Inter-model Horizontal Fusion☆32Updated 8 months ago
- Artifacts of EVT ASPLOS'24☆22Updated 10 months ago
- Set of datasets for the deep learning recommendation model (DLRM).☆41Updated 2 years ago
- ☆36Updated 2 years ago