Implementation of "Deep Graph Matching Consensus" in PyTorch
MIT License
The easiest way to use deep metric learning in your application. Modular, flexible, and extensibl...
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-f...
Pytorch framework for doing deep learning on point clouds.
PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for...
LPIPS metric. pip install lpips
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
An autoML framework & toolkit for machine learning on graphs.
Collection of SOTA efficient computer vision models for embedded applications, with pre-trained w...
PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models ...
Hierarchical Inter-Message Passing for Learning on Molecular Graphs
Constrained optimization toolkit for PyTorch
An open source implementation of CLIP.
CSSegmentation: An Open Source Continual Semantic Segmentation Toolbox Based on PyTorch.
An extension of Open3D to address 3D Machine Learning tasks