Skip to content
Machine Learning research with PyTorch
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.


Implementations of various NNs in pytorch

# simple models
from torchzoo import FeedForward, COOL

# recurrent models
from torchzoo import RWA

Benchmarks / Reproduction studies are held in TorchCircus.

How to use Zoo/Circus

  1. Finding a paper
    • all book keeping is done in the github Issues.
    • Since most papers end up on arxiv, search for an arxiv ID like 1805.10369
    • If the paper is not on Arxiv, search for the full title of the paper.
  2. The issue labels provide a mechanism to find specific things.
  3. If a paper passes it's reproduction in the Circus, the underlying CORE parts are put into the master branch.
  4. If a paper has failed in it's reproduction, the core parts are sent to the fertilizer branch.
  5. If something is still not covered, please open an issue.

I'll try to look into things whenever I have time.

You can’t perform that action at this time.