KinglittleQ / torch-batch-svd
A 100x faster SVD for PyTorch⚡️
☆470Updated 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
- ☆149Updated 2 years ago
- Vector Neurons: A General Framework for SO(3)-Equivariant Networks☆487Updated 2 years ago
- pytorch knn [cuda version]☆304Updated 3 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year
- PyTorch 1.0 implementation of the approximate Earth Mover's Distance☆138Updated 5 years ago
- Newton and Quasi-Newton optimization with PyTorch☆345Updated last year
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,091Updated last week
- ☆78Updated 4 years ago
- Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function☆486Updated last year
- PyTorch Extension Library of Optimized Scatter Operations☆1,635Updated 2 weeks ago
- RoMa: A lightweight library to deal with 3D rotations in PyTorch.☆572Updated last month
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆74Updated 2 years ago
- Geometric loss functions between point clouds, images and volumes☆622Updated 3 weeks ago
- Deep Declarative Networks☆245Updated 2 months ago
- ☆271Updated 2 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,327Updated 2 months ago
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,057Updated 3 weeks ago
- Tutorial for building a custom CUDA function for Pytorch☆511Updated 6 years ago
- Computations involving Lie groups and harmonic analysis☆202Updated 2 months ago
- ☆156Updated 9 months ago
- Learning Gradient Fields for Shape Generation☆192Updated 3 years ago
- Pytorch package to compute Chamfer distance between point sets (pointclouds).☆324Updated last year
- Preconditioned Conjugate Gradient in Pytorch☆140Updated last year
- Official Implementation of Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reduction (2020)☆199Updated 2 years ago
- Code for "Scale-Equivariant Steerable Networks"☆69Updated 4 years ago
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆865Updated 2 weeks ago
- A pytorch port of google-research/google-research/robust_loss/☆678Updated 3 years ago
- PyTorch Wrapper for Earth-Mover-Distance (EMD) for 3D point cloud regression☆296Updated 2 years ago
- tutorial for writing custom pytorch cpp+cuda kernel, applied on volume rendering (NeRF)☆406Updated 2 years ago