Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)
APACHE-2.0 License
Pytorch framework for doing deep learning on point clouds.
[CVPR2020] Adversarial Latent Autoencoders
An unofficial styleguide and best practices summary for PyTorch
Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/impleme...
The pytorch re-implement of the official efficientdet with SOTA performance in real time and pret...
Exploration of Self-Semi-Supervised learning for handling unlabeled data
Official code for the LoG2022 paper -- MSGNN: A Spectral Graph Neural Network Based on a Novel Ma...
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-e...
Experiments of Pytorch SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...
Datasets, Transforms and Models specific to Computer Vision
Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for...