borisfom

cudnn.torch

Torch-7 FFI bindings for NVIDIA CuDNN

pix2pixHD

Synthesizing and manipulating 2048x1024 images with conditional GANs

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

CMake

CMake, the cross-platform, open-source build system.

cugraph

cuGraph - RAPIDS Graph Analytics Library

DeepLearningExamples

Deep Learning Examples

e3nn

A modular framework for neural networks with Euclidean symmetry

FastPhotoStyle

NeMo

NeMo: a toolkit for conversational AI

Torch-TensorRT

PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

esm3

Lasagne

Lightweight library to build and train neural networks in Theano

libgpuarray

Library to manipulate tensors on the GPU.

nemo2riva

NeMo -> Riva Conversion Tool

neural-style

Torch implementation of neural style algorithm

onnx-tensorrt

ONNX-TensorRT: TensorRT backend for ONNX

onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

onnxscript

ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.

opencv

Open Source Computer Vision Library

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PytorchNeuralStyleTransfer

Implementation of Neural Style Transfer in Pytorch

SFD_pytorch

A PyTorch Implementation of Single Shot Scale-invariant Face Detector.

style-swap

code for Fast Patch-based Style Transfer of Arbitrary Style

styletransfer

Real-time neural style transfer via meta networks

tensorrt-inference-server

The TensorRT Inference Server provides a cloud inferencing solution optimized for NVIDIA GPUs.

texture_nets

Code for "Texture Networks: Feed-forward Synthesis of Textures and Stylized Images" paper.

Theano

Theano is a Python library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays efficiently. It can use GPUs and perform efficient symbolic differentiation.

torch7

http://torch.ch

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

VISTA

MONAI Versatile Imaging Segmentation and Annotation