- New method for atom/bond tokenizing (integer tokens, passed through torch.nn.Embedding layer at start of model)
- Removed GRU and scatter add functions, leads to faster and more accurate results
- Changed MFConv to GeneralConv (GeneralConv considers both atom and bond features during its message passing step)
- General flow for each message passing step is:
- embedded_bonds -> EdgeConv -> updated embedded_bonds
- embedded_atoms + embedded_bonds -> GeneralConv -> updated embedded_atoms
- Separate edge/node convolutions, where node convolutions also account for updated edge states, is used effectively by Jørgensen et al. ([paper](https://doi.org/10.48550/arXiv.1806.03146))
- Updated/added examples