code snippet for a PLUMED input reading ``` histo: METATOMIC ... MODEL=histo-cv.pt EXTENSIONS_DIRECTORY=./extensions/ SPECIES1=1-38 SPECIES_TO_TYPES=18 ... cv1: SELECT_COMPONENTS ARG=histo COMPONENTS=1 cv2: SELECT_COMPONENTS ARG=histo COMPONENTS=2 ```
Basically, you just need to define a `torch.nn.module` with a specific API, and then you can define anything you like as a CV calculator. Export as .pt torchscript model, and it's just on METATOMIC action away from reading it in #plumed