zou3519

dotfiles

My dotfiles

pytorch_autograd_generated

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

benchmark

builder

Continuous builder and binary build scripts for pytorch

coup

tbd

Deep-Learning

A few notebooks about deep learning in pytorch

driss_torch

Cuda extensions for PyTorch

dummy

efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

examples

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

executorch

On-device AI across mobile, embedded and edge for PyTorch

extension-cpp

C++ extensions in PyTorch

FastPhotoStyle

Style transfer, deep learning, feature transform

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

git

Git Source Code Mirror - This is a publish-only repository and all pull requests are ignored. Please follow Documentation/SubmittingPatches procedure for any of your improvements.

maml-jax

Implementation of Model-Agnostic Meta-Learning (MAML) in Jax

nestedtensor

[EXPERIMENTAL] Tools for the concurrent manipulation of variably sized Tensors.

pix2pixHD

Synthesizing and manipulating 2048x1024 images with conditional GANs

pylint

A Python source code analyzer which looks for programming errors, helps enforcing a coding standard and sniffs for some code smells

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch.github.io

pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

triton

Development repository for the Triton language and compiler

tutorials

vision

Datasets, Transforms and Models specific to Computer Vision

xla