EIDOSLAB / simplifyLinks
Simplification of pruned models for accelerated inference | SoftwareX https://doi.org/10.1016/j.softx.2021.100907
☆36Updated 7 months ago
Alternatives and similar repositories for simplify
Users that are interested in simplify are comparing it to the libraries listed below
Sorting:
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆222Updated 6 months ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆209Updated 2 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated 10 months ago
- ☆209Updated 3 years ago
- ☆75Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆217Updated 2 years ago
- [ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wa…☆33Updated 2 years ago
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆38Updated last year
- Unofficial PyTorch implementation of Google's FNet: Mixing Tokens with Fourier Transforms. With checkpoints.☆77Updated 3 years ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆42Updated 7 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆287Updated last year
- ☆164Updated 2 years ago
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆74Updated 2 years ago
- Compression schema for gradients of activations in backward pass☆44Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 2 years ago
- PyTorch Lightning + Hydra. + Timm: A very user-friendly template for rapid and reproducible MLOps with best practices. ⚡🔥⚡☆17Updated 2 years ago
- ☆51Updated last year
- Estimate/count FLOPS for a given neural network using pytorch☆306Updated 3 years ago
- ☆19Updated 3 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆141Updated 6 months ago
- A research library for pytorch-based neural network pruning, compression, and more.☆163Updated 2 years ago
- Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption☆107Updated 2 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Pytorch reimplementation of the Mixer (MLP-Mixer: An all-MLP Architecture for Vision)☆36Updated 4 years ago
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆114Updated 5 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- A simple minimal implementation of Reversible Vision Transformers☆126Updated last year
- Adaptive Gradient Clipping☆149Updated 3 years ago