415 Open source projects that are alternatives of or similar to Pytorch Unflow

A Python wrapper of NVIDIA Video Loader (NVVL) with CuPy for fast video loading with Python

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

a reimplementation of LiteFlowNet in PyTorch that matches the official Caffe version

an implementation of softmax splatting for differentiable forward warping using PyTorch

NumPy & SciPy for GPU

Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)

A flexible framework of neural networks for deep learning

an implementation of Revisiting Adaptive Convolutions for Video Frame Interpolation using PyTorch

Library for faster pinned CPU <-> GPU transfer in Pytorch

an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch

Official source code for our paper "AdaCoF: Adaptive Collaboration of Flows for Video Frame Interpolation" (CVPR 2020)

Reference implementation for "Massively Parallel Rendering of Complex Closed-Form Implicit Surfaces" (SIGGRAPH 2020)

Not All Pixels Are Equal: Difficulty-Aware Semantic Segmentation via Deep Layer Cascade

deformable convolution 2D 3D DeformableConvolution DeformConv Modulated Pytorch CUDA

ICCV2015 Image Super-Resolution Using Deep Convolutional Networks

Chamfer Distance in Pytorch with f-score

Numeric Erlang - vector and matrix operations with CUDA. Heavily inspired by Pteracuda - https://github.com/kevsmith/pteracuda

HPC solver for nonlinear optimization problems

A high-performance CUDA-based physics simulation sandbox for soft robotics and reinforcement learning.

Real-time dense visual SLAM system

CURRENNNT codes and scripts

Efficient reservoir sampling implementation for PyTorch

ChainerCV: a Library for Deep Learning in Computer Vision

PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer

Minimal Deep Learning library is written in Python/Cython/C++ and Numpy/CUDA/cuDNN.

Tutorial on how to build your own research envirorment for Deep Learning with OpenCV, Python, Tensorfow

pytorch knn [cuda version]

PyTorch 1.0 implementation of the approximate Earth Mover's Distance

DaCe - Data Centric Parallel Programming

A CUDA Extension of Neural Network Libraries

This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as recommendation and natural language processing. We showed this library can reduce the total model size by up to 100x in Facebook’s open sourced DLRM model while achieving same model quality. Our implementation is faster than the state-of-the-art implementations. Existing the state-of-the-art library also decompresses the whole embedding tables on the fly therefore they do not provide memory reduction during runtime of the training. Our library decompresses only the requested rows therefore can provide 10,000 times memory footprint reduction per embedding table. The library also includes a software cache to store a portion of the entries in the table in decompressed format for faster lookup and process.

Some CUDA design patterns and a bit of template magic for CUDA

A complete machine vision container that includes Jupyter notebooks with built-in code hinting, Anaconda, CUDA-X, TensorRT inference accelerator for Tensor cores, CuPy (GPU drop in replacement for Numpy), PyTorch, TF2, Tensorboard, and OpenCV for accelerated workloads on NVIDIA Tensor cores and GPUs.

Julia wrapper for CUDA runtime API

Ubuntu 18.04 How to install Nvidia driver + CUDA + CUDNN + build tensorflow for gpu step by step command line

An extensible HPC framework for CUDA, OpenCL and native CPU.

Fast CUDA Kernels for ResNet Inference.

The source code of ECCV16 'Deep Joint Image Filtering'.

MatConvNet: CNNs for MATLAB

GPU database engine

💥💻💥 A data-parallel functional programming language

Deep Learning library using GPU(CUDA/cuBLAS)

AutoDock for GPUs and other accelerators

Gradient Boosting powered by GPU(NVIDIA CUDA)

A Julia wrapper for the CUDA driver API.

Deep.Net machine learning framework for F#

A fast and highly scalable GPU dynamic memory allocator

GPU-Suite

@ICCV2017: For exploiting second-order statistics, we propose Matrix Power Normalized Covariance pooling (MPN-COV) ConvNets, different from and outperforming those using global average pooling.

Weighted Softmax Loss Layer for Caffe

CUDA Templates for Linear Algebra Subroutines

GGNN: State of the Art Graph-based GPU Nearest Neighbor Search

Detail-Preserving Pooling in Deep Networks (CVPR 2018)

Deep learning with cats (^._.^)

GPU Accelerated t-SNE for CUDA with Python bindings

Application of machine learning to the Coinbase (GDAX) orderbook

ThunderSVM: A Fast SVM Library on GPUs and CPUs

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