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,194Updated 2 months ago
• vahidk / tfrecord
  Standalone TFRecord reader/writer with PyTorch data loaders
  ☆897Updated 5 months ago
• peci1 / nvidia-htop
  A tool for enriching the output of nvidia-smi.
  ☆569Updated last year
• xwying / torchshow
  Visualize PyTorch tensors with a single line of code.
  ☆701Updated 2 months ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,074Updated 2 months ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆609Updated last year
• mit-han-lab / torchsparse
  [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
  ☆1,409Updated 8 months ago
• Oldpan / Pytorch-Memory-Utils
  pytorch memory track code
  ☆1,016Updated 4 years ago
• subhadarship / kmeans_pytorch
  kmeans using PyTorch
  ☆523Updated 2 years ago
• yangkky / distributed_tutorial
  ☆261Updated 6 years ago
• rusty1s / pytorch_scatter
  PyTorch Extension Library of Optimized Scatter Operations
  ☆1,700Updated 2 months ago
• fgaim / gpuview
  A lightweight web dashboard for monitoring GPU usage
  ☆203Updated last week
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆413Updated 4 months ago
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆438Updated last year
• rbgirshick / yacs
  YACS -- Yet Another Configuration System
  ☆1,325Updated 3 years ago
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆621Updated 11 months ago
• anibali / docker-pytorch
  A Docker image for PyTorch
  ☆993Updated 2 years ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆802Updated 5 months ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆629Updated 3 months ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,050Updated 4 months ago
• jeonsworld / ViT-pytorch
  Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
  ☆2,091Updated 3 years ago
• Lyken17 / Efficient-PyTorch
  My best practice of training large dataset using PyTorch.
  ☆1,107Updated last year
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆519Updated 6 years ago
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆259Updated 4 years ago
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆367Updated 7 years ago
• Lightning-AI / torchmetrics
  Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,352Updated last week
• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆570Updated 4 years ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,175Updated 2 years ago
• TylerYep / torchinfo
  View model summaries in PyTorch!
  ☆2,878Updated last week
• krasserm / perceiver-io
  A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
  ☆496Updated last year