This was part of Learning Collective Variables and Coarse Grained Models

End-to-end differentiation for reversible coarse-graining and rare-event sampling

Rafael Gomez-Bombarelli, Massachusetts Institute of Technology (MIT)