/PermutationalNetworks

Code for our paper on dynamics prediction using permutation-equivariant neural networks. The base of the repository contains an implementation of a drop-in Lasagne layer, and the subdirectories contain specific experiments from the paper.

Primary LanguagePython

PermutationalNetworks

Code from our paper on permutation equivariant networks for dynamics prediction.

The root directory of this repository contains code for a drop-in permutation equivariant layer for Lasagne. The subdirectories contain specific experiments from our paper.