ngimel

nccl.torch

torch bindings for nccl

inductor_generated

deep-photo-styletransfer

Implementation of "Deep Photo Style Transfer": https://arxiv.org/abs/1703.07511

pytorch

cutorch

A CUDA backend for Torch7

deepmark

THE Deep Learning Benchmarks

fb.resnet.torch

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts

apex

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

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

builder

Continuous builder and binary build scripts for pytorch

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

fairseq-py

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

functorch

functorch is a prototype of JAX-like composable function transforms for PyTorch.

moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

opacus

Training PyTorch models with differential privacy

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

rnn_ext

tensorflow

An Open Source Machine Learning Framework for Everyone

torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

triton

Development repository for the Triton language and compiler