ptrblck Goto Github PK

apex

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

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

deeplearningexamples

Deep Learning Examples

discogan

Webcam demo for SKTBrain/DiscoGAN

drummernet

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

gan-intro-pytorch

Intro to Generative Adversarial Networks in Pytorch

magma_win

A modified version for Windows, Current version: 2.5.4

prog_gans_pytorch_inference

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

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch.github.io

The website for PyTorch

pytorch_misc

Code snippets created for the PyTorch discussion board

talks

Slides from talks I gave

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

tutorials

vision

Datasets, Transforms and Models specific to Computer Vision