ptrblck

pytorch_misc

Code snippets created for the PyTorch discussion board

prog_gans_pytorch_inference

PyTorch inference for "Progressive Growing of GANs" with CelebA snapshot

apex

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

gan-intro-pytorch

Intro to Generative Adversarial Networks in Pytorch

DiscoGAN

Webcam demo for SKTBrain/DiscoGAN

tutorials

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

talks

Slides from talks I gave

DeepLearningExamples

Deep Learning Examples

vision

Datasets, Transforms and Models specific to Computer Vision

DrummerNet

Supplementary material of "Deep Unsupervised Drum Transcription", ISMIR 2019

audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

builder

Continuous builder and binary build scripts for pytorch

cutlass

CUDA Templates for Linear Algebra Subroutines

magma_win

A modified version for Windows, Current version: 2.5.4

pytorch.github.io

The website for PyTorch

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

test-infra

This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

text

Data loaders and abstractions for text and NLP