Alternatives and similar repositories for switch-cuda

Users that are interested in switch-cuda are comparing it to the libraries listed below

• facebookresearch / fvcore
  Collection of common code that's shared among different research projects in FAIR computer vision team.
  ☆2,203Updated 3 months ago
• vahidk / tfrecord
  Standalone TFRecord reader/writer with PyTorch data loaders
  ☆899Updated 7 months ago
• xwying / torchshow
  Visualize PyTorch tensors with a single line of code.
  ☆702Updated 3 months ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,076Updated 3 months ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆415Updated 5 months ago
• mit-han-lab / torchsparse
  [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
  ☆1,426Updated 9 months ago
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆442Updated last year
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆468Updated 2 years ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆631Updated 4 months ago
• rusty1s / pytorch_scatter
  PyTorch Extension Library of Optimized Scatter Operations
  ☆1,714Updated 4 months ago
• rbgirshick / yacs
  YACS -- Yet Another Configuration System
  ☆1,328Updated 3 years ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆611Updated last year
• KinglittleQ / torch-batch-svd
  A 100x faster SVD for PyTorch⚡️
  ☆483Updated 3 years ago
• peci1 / nvidia-htop
  A tool for enriching the output of nvidia-smi.
  ☆570Updated last year
• Oldpan / Pytorch-Memory-Utils
  pytorch memory track code
  ☆1,016Updated 4 years ago
• subhadarship / kmeans_pytorch
  kmeans using PyTorch
  ☆526Updated 2 years ago
• Lyken17 / Efficient-PyTorch
  My best practice of training large dataset using PyTorch.
  ☆1,107Updated last year
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆804Updated 6 months ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆822Updated 3 years ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆992Updated last year
• rentruewang / koila
  Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
  ☆1,832Updated last month
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆815Updated last year
• kwea123 / pytorch-cppcuda-tutorial
  tutorial for writing custom pytorch cpp+cuda kernel, applied on volume rendering (NeRF)
  ☆417Updated 2 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆520Updated 6 years ago
• SHI-Labs / Neighborhood-Attention-Transformer
  Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
  ☆1,163Updated last year
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,185Updated 2 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,053Updated 5 months ago
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆628Updated last year
• anibali / docker-pytorch
  A Docker image for PyTorch
  ☆993Updated 2 years ago
• autonomousvision / giraffe
  This repository contains the code for the CVPR 2021 paper "GIRAFFE: Representing Scenes as Compositional Generative Neural Feature Fields…
  ☆1,241Updated 3 years ago