Tutorial: Wave Architecture

Overview

In this section, we demonstrate our home grown wave architecture implemented in tflon. This example demonstrates a non trivial use case with a more complex architecture than we had previously.

Architecture

Here we implement it with a built in Graph Recurrent Neural Network (GRNN) which is a fundamental unit introduced for the construction of Wave. In short, Wave aspires to improve long range information propogation. This improvement was demonstrated to be particularly useful for operating on graphs representing molecules.

See our paper for more details.