nelaturuharsha / TurboPrune
Harness for training/finding lottery tickets in PyTorch. With support for multiple pruning techniques and augmented by distributed training, FFCV and AMP.
☆18Updated 3 months ago
Alternatives and similar repositories for TurboPrune
Users that are interested in TurboPrune are comparing it to the libraries listed below
Sorting:
- Python library for argument and configuration management☆54Updated 2 years ago
- Official code for the Stochastic Polyak step-size optimizer☆139Updated 10 months ago
- Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).☆235Updated this week
- A privacy preserving NLP framework☆198Updated 2 years ago
- implements some LRP rules to get explanations for Resnets and Densenet-121, including batchnorm-Conv canonization and tensorbiased layers…☆25Updated last year
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- Everything you need to reproduce "Better plain ViT baselines for ImageNet-1k" in PyTorch, and more☆9Updated this week
- Simple code for generating a color-coded latex table from raw data☆194Updated last year
- Exact method for visualizing partitions of a Deep Neural Network, CVPR 2023 Highlight☆104Updated 3 months ago
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆264Updated 2 years ago
- A concise primer on Differential Privacy☆29Updated 4 years ago
- Fair Generative Modeling via Weak Supervision☆20Updated 2 years ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆276Updated 2 years ago
- Learning to See by Looking at Noise☆110Updated 5 months ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- ☆133Updated 4 years ago
- Code for: Implicit Competitive Regularization in GANs☆114Updated 3 years ago
- ☆144Updated 2 years ago
- Gradient Origin Networks - a new type of generative model that is able to quickly learn a latent representation without an encoder☆161Updated 4 years ago
- Implementation of "Structured Multi-Hashing for Model Compression" (CVPR 2020)☆11Updated 4 years ago
- Code for paper "Adversarial score matching and improved sampling for image generation"☆123Updated last year
- Using federated learning and CNNs to analyze X-ray images of Lungs and determining whether pneumonia is present.☆33Updated 5 years ago
- Code for the paper "Calibrating Deep Neural Networks using Focal Loss"☆161Updated last year
- Algorithms for Privacy-Preserving Machine Learning in JAX☆94Updated last month
- The M2L school 2022 tutorials☆36Updated 2 years ago
- autodp: A flexible and easy-to-use package for differential privacy☆275Updated last year
- Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"☆479Updated 2 years ago
- Create animations for the optimization trajectory of neural nets☆153Updated last year
- Gradient based Hyperparameter Tuning library in PyTorch☆289Updated 4 years ago
- ☆57Updated last year