Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
☆437Aug 29, 2024Updated last year
Alternatives and similar repositories for toma
Users that are interested in toma are comparing it to the libraries listed below
Sorting:
- ☆14Jul 2, 2019Updated 6 years ago
- Code for the CVPR 2020 [ORAL] paper "SAM: The Sensitivity of Attribution Methods to Hyperparameters"☆27Dec 8, 2022Updated 3 years ago
- A easy to use API to store outputs from forward/backward hooks in Pytorch☆36Aug 19, 2019Updated 6 years ago
- There and Back Again: Revisiting Backpropagation Saliency Methods (CVPR 2020)☆53Apr 7, 2020Updated 5 years ago
- Run fully connected artificial neural networks with dropout applied (mini)batchwise, rather than samplewise. Given two hidden layers each…☆15May 18, 2015Updated 10 years ago
- ☆42May 20, 2020Updated 5 years ago
- A lightweight experimental logging library☆52Dec 23, 2025Updated 2 months ago
- A library for evaluating representations.☆77Nov 21, 2021Updated 4 years ago
- Estimating Example Difficulty using Variance of Gradients☆64Jan 10, 2023Updated 3 years ago
- A pytorch implementation of Progressive-GAN that is actually works, readable and simple to customize☆85Mar 12, 2022Updated 3 years ago
- Collection of machine learning research paper references☆26Feb 23, 2025Updated last year
- [CVPR 2020] Official Implementation: "Your Local GAN: Designing Two Dimensional Local Attention Mechanisms for Generative Models".☆137Nov 22, 2022Updated 3 years ago
- Semi-supervised learning via Compact Latent Space Clustering☆50Jun 6, 2019Updated 6 years ago
- ☆91Nov 15, 2019Updated 6 years ago
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Jan 1, 2024Updated 2 years ago
- Testing Nerual Tangent Kernel (NTK) on small UCI datasets☆80Oct 7, 2019Updated 6 years ago
- Code repository for the paper "Attentive Group Equivariant Convolutional Neural Networks" published at ICML 2020. https://arxiv.org/abs/2…☆53Jul 7, 2020Updated 5 years ago
- Code and Data for ACL 2020 paper "Few-Shot NLG with Pre-Trained Language Model"☆190May 23, 2025Updated 9 months ago
- Graph Neural Network☆44Oct 21, 2021Updated 4 years ago
- A PyTorch implementation of SSCR☆23Aug 12, 2024Updated last year
- JAvascript Design Environment☆88Apr 14, 2023Updated 2 years ago
- Exploring Random Encoders for Sentence Classification☆183Mar 6, 2020Updated 6 years ago
- Character-Preserving Coherent Story Visualization, ECCV 2020☆42Mar 12, 2021Updated 4 years ago
- ☆252Dec 27, 2022Updated 3 years ago
- Density-preserving data visualization tools den-SNE and densMAP☆117Apr 27, 2021Updated 4 years ago
- Totally Versatile Miscellanea for Pytorch☆477Mar 20, 2022Updated 3 years ago
- Code for Piggyback: Adapting a Single Network to Multiple Tasks by Learning to Mask Weights☆184May 2, 2019Updated 6 years ago
- Code for the Proceedings of the National Academy of Sciences 2020 article, "Understanding the Role of Individual Units in a Deep Neural N…☆307Jan 9, 2021Updated 5 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Pytorch Implementation of the paper MixMatch: A Holistic Approach to Semi-Supervised Learning (https://arxiv.org/pdf/1905.02249.pdf)☆123May 30, 2019Updated 6 years ago
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆415Jan 13, 2021Updated 5 years ago
- Mid-Level Visual Representations Improve Generalization and Sample Efficiency for Learning Visuomotor Policies☆109Mar 25, 2023Updated 2 years ago
- Web page for our tutorial on latent structure for NLP☆47Sep 1, 2019Updated 6 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆988May 30, 2025Updated 9 months ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,471May 2, 2025Updated 10 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,415Feb 20, 2026Updated 2 weeks ago
- [ICML 2021] GraphNorm: A Principled Approach to Accelerating Graph Neural Network Training (official implementation)☆107Dec 19, 2022Updated 3 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆331Jun 13, 2022Updated 3 years ago
- Code for Neural Spline Flows paper☆284Oct 19, 2025Updated 4 months ago