cupy

A flexible framework of neural networks for deep learning

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

A simplified implemention of Faster R-CNN that replicate performance from origin paper

ChainerCV: a Library for Deep Learning in Computer Vision

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

TensorLy: Tensor Learning in Python.

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

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

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

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

an implementation of softmax splatting for differentiable forward warping using PyTorch

CuPy fused PyTorch neural networks ops

Deep learning framework realized by Numpy purely, supports for both Dynamic Graph and Static Graph with GPU acceleration

A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python 🚀

A Simple & Flexible Cross Framework Operators Toolkit

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

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

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

Chainer implementation of waifu2x

a reimplementation of UnFlow in PyTorch that matches the official TensorFlow version

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors

The code for multi-channel source separation and dereverberation such as FastMNMF1, FastMNMF2, and AR-FastMNMF2.

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

Minimal deep learning library written from scratch in Python, using NumPy/CuPy.

🔬 Scientific computing library for optics 🔭, computer graphics 💻 and visual perception 👀

ROIPooling for pytorch

This container is no longer supported, and has been deprecated in favor of: https://github.com/joehoeller/NVIDIA-GPU-Tensor-Core-Accelerator-PyTorch-OpenCV

Write numeric code that automatically works with any numpy-ish libraries.