steveli / pytorch-sqrtmLinks
Matrix square root with gradient support for PyTorch
☆69Updated 3 years ago
Alternatives and similar repositories for pytorch-sqrtm
Users that are interested in pytorch-sqrtm are comparing it to the libraries listed below
Sorting:
- Backpropagation-Friendly-Eigendecomposition☆73Updated 5 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆236Updated 4 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
- Deep Isometric Learning for Visual Recognition (ICML 2020)☆143Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆72Updated 5 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated 2 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- Official implementation of "Energy-Based Models for Deep Probabilistic Regression" (ECCV 2020) and "How to Train Your Energy-Based Model …☆105Updated 5 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 4 years ago
- Code of Empirical Bayes Transductive Meta-Learning with Synthetic Gradients☆58Updated 5 years ago
- Pytorch Implementation of OpenAI's GLOW☆94Updated 4 years ago
- implements optimal transport algorithms in pytorch☆101Updated 3 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆119Updated 4 years ago
- Official repository for the paper "Masksembles for Uncertainty Estimation" (CVPR 2021).☆102Updated 4 months ago
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆129Updated 3 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆129Updated 2 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆67Updated 4 years ago
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆156Updated last year
- This is the pytorch re-implementation of the IterNorm☆41Updated 6 years ago
- Attentive Single-tasking of Multiple Tasks☆84Updated 6 years ago
- Percentile computation for pytorch☆21Updated 5 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- Sliced Wasserstein Generator☆39Updated 7 years ago
- Implementation of the reversible residual network in pytorch☆105Updated 3 years ago
- Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problem [CVPR 2019, oral]☆185Updated 6 years ago
- PyTorch implementation of "Glow: Generative Flow with Invertible 1x1 Convolutions"☆45Updated 7 years ago
- Pytorch implementation of Learning Rate Dropout.☆42Updated 5 years ago
- (CVPR 2020) This repo contains code for "PADS: Policy-Adapted Sampling for Visual Similarity Learning", which proposes learnable triplet …☆63Updated 5 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated 2 years ago