awslabs / lorienLinks
☆43Updated last year
Alternatives and similar repositories for lorien
Users that are interested in lorien are comparing it to the libraries listed below
Sorting:
- ☆23Updated 6 months ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆121Updated 2 years ago
- ☆143Updated 4 months ago
- DietCode Code Release☆64Updated 2 years ago
- ☆92Updated 2 years ago
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆25Updated 3 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆138Updated 2 years ago
- Cavs: An Efficient Runtime System for Dynamic Neural Networks☆14Updated 4 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆50Updated 10 months ago
- Repository for SysML19 Artifacts Evaluation☆54Updated 6 years ago
- Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation☆27Updated 5 years ago
- Benchmark PyTorch Custom Operators☆14Updated last year
- An IR for efficiently simulating distributed ML computation.☆28Updated last year
- ☆47Updated 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 …☆65Updated 3 years ago
- ☆79Updated 2 years ago
- This is the implementation for paper: AdaTune: Adaptive Tensor Program CompilationMade Efficient (NeurIPS 2020).☆13Updated 4 years ago
- PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.☆10Updated 3 years ago
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆76Updated 4 years ago
- FTPipe and related pipeline model parallelism research.☆41Updated 2 years ago
- Boost hardware utilization for ML training workloads via Inter-model Horizontal Fusion☆32Updated last year
- A schedule language for large model training☆148Updated 11 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆131Updated 3 years ago
- Benchmark scripts for TVM☆74Updated 3 years ago
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆24Updated 4 years ago
- ☆16Updated 2 years ago
- Artifacts of EVT ASPLOS'24☆25Updated last year
- An external memory allocator example for PyTorch.☆14Updated 3 years ago
- My paper/code reading notes in Chinese☆46Updated last year