aliutkus / torchsearchsorted
Pytorch Custom CUDA kernel for searchsorted
☆136Updated last year
Alternatives and similar repositories for torchsearchsorted:
Users that are interested in torchsearchsorted are comparing it to the libraries listed below
- Preconditioned Conjugate Gradient in Pytorch☆138Updated last year
- Interpolation routines in Pytorch.☆110Updated 4 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆232Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆67Updated 4 years ago
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆263Updated 2 years ago
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields☆67Updated last year
- CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuou…☆48Updated 3 years ago
- PyTorch implementation of cvpr2019 paper "Spherical Regression: Learning Viewpoints, Surface Normals and 3D Rotations on n-Spheres".☆147Updated 3 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago
- Implementation of Differentiable Sign-Distance Function Rendering - in Pytorch☆68Updated 2 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated last month
- ☆79Updated 3 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
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- ☆77Updated 3 years ago
- Pytorch Implementation of OpenAI's GLOW☆93Updated 3 years ago
- Official code for "On the Frequency Bias of Generative Models", NeurIPS 2021☆44Updated 3 years ago
- Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function☆479Updated last year
- Pytorch implementation of Generative Models as Distributions of Functions 🌿☆148Updated last year
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆92Updated 3 years ago
- Fast analytical implementation of batch eigen-decomposition for 3x3 symmetric matrices with Pytorch. > 250x faster than regular Pytorch i…☆21Updated 4 years ago
- DDSL: Deep Differential Simplex Layer for Neural Networks☆55Updated 4 years ago
- Learning Gradient Fields for Shape Generation☆189Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 9 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- (Batched) advanced indexing for PyTorch.☆53Updated last month
- Non-official + minimal reimplementation of HoloGAN by Nguyen-Phuoc, et al: https://arxiv.org/abs/1904.01326☆65Updated 4 years ago
- code for "Residual Flows for Invertible Generative Modeling".☆265Updated last year
- ☆164Updated 3 years ago