aliutkus / torchsearchsorted
Pytorch Custom CUDA kernel for searchsorted
☆137Updated last year
Alternatives and similar repositories for torchsearchsorted:
Users that are interested in torchsearchsorted are comparing it to the libraries listed below
- Interpolation routines in Pytorch.☆110Updated 4 years ago
- CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuou…☆50Updated 3 years ago
- Preconditioned Conjugate Gradient in Pytorch☆140Updated last year
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 4 months ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆234Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆69Updated 4 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆264Updated 2 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆255Updated last year
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- ☆78Updated 4 years ago
- PyTorch implementation of cvpr2019 paper "Spherical Regression: Learning Viewpoints, Surface Normals and 3D Rotations on n-Spheres".☆148Updated 3 years ago
- Implementation of Differentiable Sign-Distance Function Rendering - in Pytorch☆68Updated 2 years ago
- Differentiable bitonic sorting☆140Updated 4 years ago
- Gradient Origin Networks - a new type of generative model that is able to quickly learn a latent representation without an encoder☆161Updated 4 years ago
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields☆68Updated 2 years ago
- Code for UAI 2020 paper "Locally Masked Convolution for Autoregressive Models"☆77Updated 4 years ago
- Matrix square root with gradient support for PyTorch☆67Updated 3 years ago
- Pytorch implementation of Generative Models as Distributions of Functions 🌿☆148Updated last year
- code for "Residual Flows for Invertible Generative Modeling".☆269Updated last year
- Masked Convolutional Flow☆59Updated 5 years ago
- Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function☆486Updated last year
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆94Updated 4 years ago
- Pytorch Implementation of OpenAI's GLOW☆94Updated 4 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆178Updated 3 years ago
- An unofficial pytorch implementation of 《Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains》☆66Updated 4 years ago
- Percentile computation for pytorch☆21Updated 5 years ago
- ☆80Updated 3 years ago
- Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'☆52Updated 4 years ago