aliutkus / torchsearchsortedLinks
Pytorch Custom CUDA kernel for searchsorted
☆137Updated 2 years ago
Alternatives and similar repositories for torchsearchsorted
Users that are interested in torchsearchsorted are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆264Updated 2 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆236Updated 4 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 4 years ago
- Interpolation routines in Pytorch.☆111Updated 4 years ago
- Code for "Scale-Equivariant Steerable Networks"☆72Updated 5 years ago
- Backpropagation-Friendly-Eigendecomposition☆73Updated 5 years ago
- Implementation of Differentiable Sign-Distance Function Rendering - in Pytorch☆69Updated 3 years ago
- PyTorch 1.0 implementation of the approximate Earth Mover's Distance☆140Updated 6 years ago
- Preconditioned Conjugate Gradient in Pytorch☆140Updated last year
- Implementation of π-GAN, for 3d-aware image synthesis, in Pytorch☆121Updated 4 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆119Updated 5 months ago
- Matrix square root with gradient support for PyTorch☆69Updated 3 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆255Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields☆68Updated 2 years ago
- Code for Scalable Uncertainty for Computer Vision with Functional Variational Inference @ CVPR 2020☆68Updated 5 years ago
- Non-official + minimal reimplementation of HoloGAN by Nguyen-Phuoc, et al: https://arxiv.org/abs/1904.01326☆65Updated 5 years ago
- A second-order optimizer for deep networks☆25Updated 5 years ago
- PyTorch implementation of cvpr2019 paper "Spherical Regression: Learning Viewpoints, Surface Normals and 3D Rotations on n-Spheres".☆149Updated 4 years ago
- Gradient Origin Networks - a new type of generative model that is able to quickly learn a latent representation without an encoder☆159Updated 4 years ago
- An unofficial pytorch implementation of 《Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains》☆67Updated 5 years ago
- Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuo…☆92Updated 6 years ago
- Official PyTorch implementation of 'RELATE: Physically Plausible Multi-Object SceneSynthesis Using Structured Latent Spaces'.☆31Updated 4 years ago
- ☆89Updated last year
- Code for UAI 2020 paper "Locally Masked Convolution for Autoregressive Models"☆79Updated 5 years ago
- Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'☆52Updated 5 years ago
- DDSL: Deep Differential Simplex Layer for Neural Networks☆55Updated 4 years ago
- ☆79Updated 4 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago