spcl / daceml
A Data-Centric Compiler for Machine Learning
☆82Updated last year
Alternatives and similar repositories for daceml:
Users that are interested in daceml are comparing it to the libraries listed below
- A lightweight, Pythonic, frontend for MLIR☆81Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆135Updated 2 years ago
- Data-Centric MLIR dialect☆40Updated last year
- MLIR-based partitioning system☆80Updated this week
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆80Updated last week
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆85Updated 2 years ago
- ☆94Updated last year
- GEMM and Winograd based convolutions using CUTLASS☆26Updated 4 years ago
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆51Updated last year
- PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.☆108Updated 4 months ago
- ☆93Updated last week
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated last year
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆40Updated last month
- ☆71Updated 3 years ago
- A schedule language for large model training☆145Updated 10 months ago
- Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018☆71Updated 4 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆105Updated 9 months ago
- ☆50Updated last year
- LLM Inference analyzer for different hardware platforms☆62Updated 2 weeks ago
- ☆77Updated 5 months ago
- ☆78Updated 2 years ago
- extensible collectives library in triton☆85Updated 2 weeks ago
- ☆23Updated 4 months ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆62Updated 3 weeks ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆99Updated last month
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆37Updated 8 months ago
- TileFlow is a performance analysis tool based on Timeloop for fusion dataflows☆57Updated last year
- ☆38Updated 5 years ago
- Re-implementation of the TASO compiler using equality saturation☆124Updated 3 years ago
- The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.☆89Updated this week