[nn] add option to take a torch.nn.module to build a nncv

luigibonati / mlcolvar

A unified framework for machine learning collective variables for enhanced sampling simulations

MIT License

90 stars 25 forks source link

Closed luigibonati closed 1 year ago