HazyResearch / butterfly

Related projects ⓘ

Alternatives and complementary repositories for butterfly

• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆148Updated 3 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆67Updated last year
• HazyResearch / fly
  ☆194Updated last year
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• google / aqt
  ☆267Updated last week
• albanD / subclass_zoo
  ☆148Updated 5 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆187Updated 6 months ago
• Tiiiger / QPyTorch
  Low Precision Arithmetic Simulation in PyTorch
  ☆265Updated 6 months ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆179Updated 3 months ago
• google-research / jaxpruner
  ☆207Updated 6 months ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆249Updated 3 years ago
• gpauloski / kfac-pytorch
  Distributed K-FAC Preconditioner for PyTorch
  ☆80Updated this week
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆127Updated last month
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆105Updated 2 weeks ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• NVlabs / condensa
  Programmable Neural Network Compression
  ☆147Updated 2 years ago
• stevenygd / SWALP
  Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".
  ☆62Updated 5 years ago
• BradMcDanel / sdgp
  ☆11Updated 2 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆160Updated last year
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆343Updated 3 weeks ago
• uber-research / deconstructing-lottery-tickets
  ☆143Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆129Updated 2 years ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆245Updated this week
• topal-team / rockmate
  ☆33Updated last year
• cybertronai / pytorch-sso
  PyTorch-SSO: Scalable Second-Order methods in PyTorch
  ☆142Updated last year
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆163Updated 3 months ago
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆49Updated 6 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆125Updated 3 years ago