Sparsity support for PyTorch
☆38Mar 22, 2025Updated 11 months ago
Alternatives and similar repositories for sten
Users that are interested in sten are comparing it to the libraries listed below
Sorting:
- Main repo to keep scripts, dockerfiles, wiki, etc☆15Mar 14, 2023Updated 3 years ago
- Compressing weather and climate data into neural networks☆17May 19, 2024Updated last year
- Deep renormalized Mori-Zwanzig (DrMZ) Julia package.☆17Jun 1, 2023Updated 2 years ago
- A Sparse-tensor Communication Framework for Distributed Deep Learning☆13Nov 1, 2021Updated 4 years ago
- FPGA-based HyperLogLog Accelerator☆12Jul 13, 2020Updated 5 years ago
- A library for code transformations with guaranteed legality☆18Updated this week
- CRISPR, faster, better – The Crackling method for whole-genome target detection☆11Jan 11, 2024Updated 2 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆30Feb 4, 2025Updated last year
- ☆32Aug 24, 2022Updated 3 years ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆237Sep 24, 2023Updated 2 years ago
- Flexible local Fourier analysis library.☆12Jun 22, 2021Updated 4 years ago
- ☆30Nov 23, 2023Updated 2 years ago
- See if we can't do some real-time learning for GMRES -- Rejoice!☆12Jun 19, 2022Updated 3 years ago
- A Data-Centric Compiler for Machine Learning☆85Dec 14, 2025Updated 3 months ago
- Sparse kernels for GNNs based on TVM☆17Nov 18, 2020Updated 5 years ago
- A Winograd Minimal Filter Implementation in CUDA☆28Aug 25, 2021Updated 4 years ago
- Source code for the CPU-Free model - a fully autonomous execution model for multi-GPU applications that completely excludes the involveme…☆22Apr 25, 2024Updated last year
- Data structures for graph neural network☆18May 25, 2024Updated last year
- GPTQ inference TVM kernel☆40Apr 25, 2024Updated last year
- ☆14Jul 18, 2019Updated 6 years ago
- The CaMLSys project template used for researching Federated Learning.☆23Updated this week
- Org babel functions for prolog evaluation.☆13Nov 4, 2019Updated 6 years ago
- This is the repository for the code used in the ICML23 paper called "Achieving High Accuracy with PINNs via Energy Natural Gradient Desce…☆28Oct 22, 2024Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆293Mar 10, 2026Updated last week
- ☆30Jun 7, 2023Updated 2 years ago
- Material for the Paper Reading sessions organized by Machine Learning Tokyo☆15Nov 5, 2019Updated 6 years ago
- ☆14Mar 11, 2024Updated 2 years ago
- Benchmark PyTorch Custom Operators☆14Jul 6, 2023Updated 2 years ago
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆17Mar 13, 2023Updated 3 years ago
- An intuitive open-source framework specific to radical pair spin dynamics.☆17Mar 6, 2026Updated 2 weeks ago
- ☆166Jul 22, 2024Updated last year
- PyTorch codes for the paper "An Empirical Study of Multimodal Model Merging"☆37Oct 11, 2023Updated 2 years ago
- High-Performance Sparse Linear Algebra on HBM-Equipped FPGAs Using HLS☆95Sep 27, 2024Updated last year
- Source code for Deep Multigrid method https://arxiv.org/pdf/1711.03825.pdf☆19May 3, 2025Updated 10 months ago
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆23Aug 21, 2020Updated 5 years ago
- Hop-Wise Graph Attention for Scalable and Generalizable Learning on Circuits☆35Aug 25, 2024Updated last year
- ☆14Aug 29, 2023Updated 2 years ago
- ☆57Jun 29, 2025Updated 8 months ago
- GPU implementation of Winograd convolution☆10Oct 23, 2017Updated 8 years ago