KinglittleQ / torch-batch-svd
A 100x faster SVD for PyTorch⚡️
☆459Updated 2 years ago
Alternatives and similar repositories for torch-batch-svd:
Users that are interested in torch-batch-svd are comparing it to the libraries listed below
- Vector Neurons: A General Framework for SO(3)-Equivariant Networks☆473Updated 2 years ago
- pytorch knn [cuda version]☆300Updated 3 years ago
- ☆146Updated 2 years ago
- Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function☆479Updated last year
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,267Updated 3 months ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆232Updated 3 years ago
- Geometric loss functions between point clouds, images and volumes☆614Updated last year
- Pytorch Custom CUDA kernel for searchsorted☆136Updated last year
- ☆77Updated 3 years ago
- PyTorch Wrapper for Earth-Mover-Distance (EMD) for 3D point cloud regression☆292Updated last year
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- PyTorch 1.0 implementation of the approximate Earth Mover's Distance☆138Updated 5 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,073Updated this week
- Learning Gradient Fields for Shape Generation☆189Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆67Updated 4 years ago
- ☆262Updated last year
- Group Equivariant Convolutional Neural Networks☆99Updated 7 years ago
- Pytorch package to compute Chamfer distance between point sets (pointclouds).☆312Updated 10 months ago
- Implementation of the Chamfer Distance as a module for pyTorch☆287Updated last year
- Chamfer Distance in Pytorch with f-score☆349Updated 4 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆217Updated last year
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆137Updated 3 years ago
- ☆716Updated last year
- Computations involving Lie groups and harmonic analysis☆197Updated last week
- E(2)-Equivariant CNNs Library for Pytorch☆627Updated last year
- Newton and Quasi-Newton optimization with PyTorch☆335Updated 11 months ago
- Coder for "On the Continuity of Rotation Representations"☆347Updated 4 years ago
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆74Updated 2 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- Tutorial for building a custom CUDA function for Pytorch☆513Updated 6 years ago