Alternatives and similar repositories for efficientnet-pytorch:

Users that are interested in efficientnet-pytorch are comparing it to the libraries listed below

• BobMcDear / vit-pytorch
  PyTorch implementation of the vision transformer
  ☆18Updated 2 years ago
• BobMcDear / neural-network-cuda
  Neural network from scratch in CUDA/C++
  ☆78Updated 3 months ago
• BobMcDear / attention-in-vision
  PyTorch implementation of popular attention mechanisms in vision
  ☆17Updated 2 years ago
• BobMcDear / simsiam-pytorch
  PyTorch implementation of SimSiam
  ☆8Updated 2 years ago
• zzd1992 / FlashWindowAttention
  Speedup the attention computation of Swin Transformer
  ☆13Updated 3 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆125Updated 4 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆534Updated this week
• aiha-lab / Attention-Head-Pruning
  Layer-wise Pruning of Transformer Heads for Efficient Language Modeling
  ☆21Updated 3 years ago
• hkproj / triton-flash-attention
  ☆157Updated 3 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated 8 months ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆213Updated 2 years ago
• BobMcDear / flaim
  Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
  ☆19Updated last year
• devansh20la / LPF-SGD
  ☆16Updated 2 years ago
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆179Updated last year
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆40Updated 2 months ago
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆112Updated 6 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆174Updated 11 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆200Updated this week
• VITA-Group / SLaK
  [ICLR 2023] "More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsity"; [ICML 2023] "Are Large Kernels Better Teachers…
  ☆267Updated last year
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆217Updated 3 months ago
• thomasverelst / awesome-dynamic-conditional-networks-cv
  Overview of conditional computution and dynamic CNNs for computer vision, with a focus on reducing computational complexity
  ☆42Updated 2 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆160Updated 2 years ago
• LukasHedegaard / pytorch-benchmark
  Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
  ☆101Updated last year
• hunto / image_classification_sota
  Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.
  ☆82Updated last year
• ehuynh1106 / TinyImageNet-Transformers
  Transformers trained on Tiny ImageNet
  ☆54Updated 2 years ago
• gpu-mode / profiling-cuda-in-torch
  ☆154Updated last year
• aanna0701 / SPT_LSA_ViT
  Implementation of Visual Transformer for Small-size Datasets
  ☆121Updated 3 years ago
• tspeterkim / mixed-precision-from-scratch
  Mixed precision training from scratch with Tensors and CUDA
  ☆22Updated 11 months ago
• hkproj / quantization-notes
  Notes on quantization in neural networks
  ☆80Updated last year
• mlops-discord / gpu-optimization-workshop
  Slides, notes, and materials for the workshop
  ☆324Updated 10 months ago