xwying / torchshow
Visualize PyTorch tensors with a single line of code.
☆664Updated last month
Alternatives and similar repositories for torchshow:
Users that are interested in torchshow are comparing it to the libraries listed below
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,429Updated last week
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆577Updated 5 months ago
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,482Updated last week
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆428Updated 7 months ago
- Tensors, for human consumption☆1,249Updated 5 months ago
- YACS -- Yet Another Configuration System☆1,311Updated 3 years ago
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆559Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆789Updated 10 months ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆218Updated last month
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆815Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 2 years ago
- ☆458Updated 2 years ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,258Updated last week
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,925Updated 10 months ago
- Profiling and inspecting memory in pytorch☆1,057Updated 9 months ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,125Updated 5 months ago
- Code to reproduce the results in the FAIR research papers "Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting V…☆488Updated 2 years ago
- Learning Rate Warmup in PyTorch☆410Updated last month
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,071Updated 2 years ago
- [NeurIPS 2022] Official code for "Focal Modulation Networks"☆730Updated last year
- View model summaries in PyTorch!☆2,771Updated last week
- A collection of literature after or concurrent with Masked Autoencoder (MAE) (Kaiming He el al.).☆822Updated 9 months ago
- Probing the representations of Vision Transformers.☆324Updated 2 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,115Updated 11 months ago
- Hiera: A fast, powerful, and simple hierarchical vision transformer.☆979Updated last year
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆476Updated 2 years ago
- iBOT : Image BERT Pre-Training with Online Tokenizer (ICLR 2022)☆721Updated 3 years ago
- Implementation of Slot Attention from GoogleAI☆426Updated 8 months ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆583Updated 6 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,424Updated this week