BobMcDear / efficientnet-pytorch
PyTorch implementation of EfficientNet
☆10Updated 2 years ago
Alternatives and similar repositories for efficientnet-pytorch:
Users that are interested in efficientnet-pytorch are comparing it to the libraries listed below
- Neural network from scratch in CUDA/C++☆78Updated 2 months ago
- PyTorch implementation of the vision transformer☆18Updated 2 years ago
- PyTorch implementation of popular attention mechanisms in vision☆15Updated last year
- PyTorch implementation of SimSiam☆8Updated 2 years ago
- Transformers trained on Tiny ImageNet☆54Updated 2 years ago
- Fast and memory-efficient exact attention☆17Updated 8 months ago
- [ICLR 2023] "More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsity"; [ICML 2023] "Are Large Kernels Better Teachers…☆266Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆524Updated last month
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆39Updated last month
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆212Updated 2 years ago
- [BMVC 2022] Official repository for "How to Train Vision Transformer on Small-scale Datasets?"☆149Updated last year
- ☆16Updated 2 years ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆51Updated 11 months ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- Speedup the attention computation of Swin Transformer☆10Updated 2 months ago
- Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption☆98Updated last year
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆308Updated 3 months ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆174Updated last year
- Compare neural networks by their feature similarity☆356Updated last year
- ☆152Updated last year
- [ICCV'23] Official repository of paper SwiftFormer: Efficient Additive Attention for Transformer-based Real-time Mobile Vision Applicatio…☆274Updated last year
- Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.☆82Updated last year
- A library for researching neural networks compression and acceleration methods.☆141Updated 7 months ago
- Simple CIFAR-10 classification with ConvMixer☆43Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆154Updated 2 years ago
- Official PyTorch implementation of the paper entitled 'Self Attentive Pooling for Efficient Deep Learning'.☆10Updated 10 months ago
- [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention☆833Updated last week
- ☆142Updated 2 months ago
- [ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmen…☆467Updated last year