ML-MD A collection of jupyter notebooks using GNN models for MD simulation and cheminformatic tasks. Trajectory prediction comparison torch baseline model vs e3NN equivariant network pytorch baseline model e3NN model