KinglittleQ / torch-batch-svd

Related projects: ⓘ

• FlyingGiraffe / vnn
  Vector Neurons: A General Framework for SO(3)-Equivariant Networks
  ☆452Updated last year
• lucidrains / siren-pytorch
  Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function
  ☆465Updated last year
• fwilliams / scalable-pytorch-sinkhorn
  ☆131Updated last year
• jeanfeydy / geomloss
  Geometric loss functions between point clouds, images and volumes
  ☆582Updated 8 months ago
• locuslab / mdeq
  [NeurIPS'20] Multiscale Deep Equilibrium Models
  ☆232Updated 3 years ago
• mit-han-lab / torchsparse
  [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
  ☆1,186Updated last month
• anucvml / ddn
  Deep Declarative Networks
  ☆236Updated 4 months ago
• daerduoCarey / PyTorchEMD
  PyTorch Wrapper for Earth-Mover-Distance (EMD) for 3D point cloud regression
  ☆284Updated last year
• princeton-vl / lietorch
  ☆673Updated 10 months ago
• unlimblue / KNN_CUDA
  pytorch knn [cuda version]
  ☆283Updated 2 years ago
• aliutkus / torchsearchsorted
  Pytorch Custom CUDA kernel for searchsorted
  ☆136Updated 10 months ago
• mfinzi / LieConv
  ☆257Updated last year
• RuojinCai / ShapeGF
  Learning Gradient Fields for Shape Generation
  ☆186Updated 3 years ago
• meder411 / PyTorch-EMDLoss
  PyTorch 1.0 implementation of the approximate Earth Mover's Distance
  ☆134Updated 5 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆515Updated 5 years ago
• krrish94 / chamferdist
  Pytorch package to compute Chamfer distance between point sets (pointclouds).
  ☆296Updated 5 months ago
• adambielski / GrouPy
  Group Equivariant Convolutional Neural Networks
  ☆96Updated 6 years ago
• chrdiller / pyTorchChamferDistance
  Implementation of the Chamfer Distance as a module for pyTorch
  ☆276Updated last year
• QUVA-Lab / e2cnn
  E(2)-Equivariant CNNs Library for Pytorch
  ☆597Updated 11 months ago
• AMLab-Amsterdam / lie_learn
  Computations involving Lie groups and harmonic analysis
  ☆190Updated last month
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,038Updated 2 months ago
• naver / roma
  RoMa: A lightweight library to deal with 3D rotations in PyTorch.
  ☆499Updated 2 months ago
• rfeinman / pytorch-minimize
  Newton and Quasi-Newton optimization with PyTorch
  ☆306Updated 6 months ago
• patrick-kidger / torchcubicspline
  Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…
  ☆215Updated last year
• tensorfieldnetworks / tensorfieldnetworks
  ☆151Updated 4 years ago
• ThibaultGROUEIX / ChamferDistancePytorch
  Chamfer Distance in Pytorch with f-score
  ☆324Updated 3 years ago
• luost26 / diffusion-point-cloud
  Diffusion Probabilistic Models for 3D Point Cloud Generation (CVPR 2021)
  ☆629Updated last year
• otaheri / chamfer_distance
  Implementation of the Chamfer Distance as a module for PyTorch
  ☆161Updated last year
• stevenygd / PointFlow
  PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows
  ☆720Updated last month
• jonbarron / robust_loss_pytorch
  A pytorch port of google-research/google-research/robust_loss/
  ☆658Updated 3 years ago