pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch

MIT License

Downloads
6.1K
Stars
165
Committers
8
View Code on GitHub Visit Website View on X
Ecosystems: PyTorch

Bot releases are visible (Hide)

pytorch_spline_conv - 1.2.2 Latest Release

Published by rusty1s over 1 year ago

  • Added support for PyTorch 2.0
pytorch_spline_conv - 1.2.1

Published by rusty1s over 3 years ago

  • Reduced the size of shared library files
  • CUDA wheels can now also operate on CPU-only devices
pytorch_spline_conv - 1.2.0

Published by rusty1s over 4 years ago

This release introduces a fully-traceable SplineConv operator using torch.jit.script. As a result, PyTorch 1.4 is now required to install this package. We also provide Python wheels for common OS/CUDA combinations (see README).

pytorch_spline_conv - 1.1.1

Published by rusty1s about 5 years ago

This release brings PyTorch 1.3 support to torch-spline-conv.

pytorch_spline_conv - 1.1.0

Published by rusty1s over 5 years ago

  • PyTorch 1.1.0
pytorch_spline_conv - 1.0.6

Published by rusty1s over 5 years ago

  • multi-GPU support
pytorch_spline_conv - 1.0.5

Published by rusty1s over 5 years ago

pytorch_spline_conv - 1.0.4

Published by rusty1s about 6 years ago

  • now uses new PyTorch extension API
  • PyTorch 0.4.1 compatibility
pytorch_spline_conv - 1.0.3

Published by rusty1s over 6 years ago

Added a norm argument: Whether to normalize output by node degree.

pytorch_spline_conv - 1.0.1

Published by rusty1s over 6 years ago

PyTorch 0.4.0

pytorch_spline_conv - 0.1.0

Published by rusty1s over 6 years ago

Package Rankings
Top 3.03% on Pypi.org
Top 15.52% on Spack.io
Badges
Extracted from project README
PyPI Version Testing Status Linting Status Code Coverage
Related Projects
pytorch_geometric_signed_directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for...

28 Oct 2021 116
pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

15 Mar 2019 1,476
torchcubicspline

Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, ...

09 Mar 2020 217
deltaconv

Code for the SIGGRAPH 2022 paper "DeltaConv: Anisotropic Operators for Geometric Deep Learning on...

12 May 2022 135
torch-points3d

Pytorch framework for doing deep learning on point clouds.

06 Dec 2019 2,463
pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models ...

27 Jun 2020 2,629
pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

16 Dec 2017 1,460
pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

28 Jul 2018 939
Pointcept

Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT...

21 Mar 2023 1,143
TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

11 Mar 2020 2,359
torchdrug

A powerful and flexible machine learning platform for drug discovery

10 Aug 2021 1,427
pytorch_cluster

PyTorch Extension Library of Optimized Graph Cluster Algorithms

12 Jan 2018 752
MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

04 Apr 2019 2,344
DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reprod...

02 May 2018 13,307
torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

12 Aug 2019 568