HazyResearch / butterfly

Related projects: ⓘ

• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆147Updated 3 years ago
• gpauloski / kfac-pytorch
  Distributed K-FAC Preconditioner for PyTorch
  ☆75Updated last week
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆65Updated 11 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆183Updated 3 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆105Updated this week
• HazyResearch / fly
  ☆187Updated last year
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆103Updated 2 years ago
• google-research / jaxpruner
  ☆202Updated 4 months ago
• google / aqt
  ☆248Updated this week
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆176Updated last month
• stevenygd / SWALP
  Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".
  ☆62Updated 5 years ago
• Tiiiger / QPyTorch
  Low Precision Arithmetic Simulation in PyTorch
  ☆258Updated 3 months ago
• plumerai / rethinking-bnn-optimization
  Implementation for the paper "Latent Weights Do Not Exist: Rethinking Binarized Neural Network Optimization"
  ☆73Updated 4 years ago
• mil-ad / studying-binary-neural-networks
  ☆53Updated 5 years ago
• google-research / rigl
  End-to-end training of sparse deep neural networks with little-to-no performance loss.
  ☆315Updated last year
• uber-research / deconstructing-lottery-tickets
  ☆142Updated last year
• fsschneider / DeepOBS
  DeepOBS: A Deep Learning Optimizer Benchmark Suite
  ☆103Updated 8 months ago
• amirgholami / HessianFlow
  ☆81Updated 4 years ago
• NVlabs / condensa
  Programmable Neural Network Compression
  ☆146Updated 2 years ago
• topal-team / rockmate
  ☆32Updated last year
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆94Updated 2 weeks ago
• IST-DASLab / WoodFisher
  Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)
  ☆45Updated 3 years ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆328Updated this week
• chrundle / biprop
  Identify a binary weight or binary weight and activation subnetwork within a randomly initialized network by only pruning and binarizing …
  ☆48Updated 2 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆161Updated last year
• wangmaolin / niti
  Implementation of "NITI: Training Integer Neural Networks Using Integer-only Arithmetic" on arxiv
  ☆75Updated 2 years ago
• spcl / sten
  Sparsity support for PyTorch
  ☆31Updated 11 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆127Updated 2 years ago
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 3 years ago