phohenecker / switch-cudaLinks
A simple bash script for switching between installed versions of CUDA.
☆646Updated 6 years ago
Alternatives and similar repositories for switch-cuda
Users that are interested in switch-cuda are comparing it to the libraries listed below
Sorting:
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,168Updated last week
- PyTorch Extension Library of Optimized Scatter Operations☆1,685Updated 3 weeks ago
- A Docker image for PyTorch☆995Updated 2 years ago
- Standalone TFRecord reader/writer with PyTorch data loaders☆890Updated 3 months ago
- Visualize PyTorch tensors with a single line of code.☆685Updated this week
- kmeans using PyTorch☆515Updated 2 years ago
- Profiling and inspecting memory in pytorch☆1,066Updated last year
- A tool for enriching the output of nvidia-smi.☆566Updated last year
- This is a pytorch implementation of k-means clustering algorithm☆323Updated 5 months ago
- ☆465Updated 2 years ago
- Code snippets created for the PyTorch discussion board☆571Updated 4 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆435Updated 11 months ago
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,387Updated 6 months ago
- YACS -- Yet Another Configuration System☆1,319Updated 3 years ago
- An All-MLP solution for Vision, from Google AI☆1,037Updated last month
- View model summaries in PyTorch!☆2,847Updated this week
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆603Updated 10 months ago
- Learning Rate Warmup in PyTorch☆410Updated 2 months ago
- My best practice of training large dataset using PyTorch.☆1,101Updated last year
- Pytorch Lightning code guideline for conferences☆1,279Updated last year
- Demystify RAM Usage in Multi-Process Data Loaders☆197Updated 2 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,170Updated 2 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆795Updated last year
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆486Updated last year
- A lightweight web dashboard for monitoring GPU usage☆201Updated last year
- A 100x faster SVD for PyTorch⚡️☆475Updated 2 years ago
- pytorch memory track code☆1,018Updated 4 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆601Updated 9 months ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆819Updated 3 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆626Updated last month