some tools for working with protein (PDB) files in tensorflow - View it on GitHub