enerzyme.models.cutoff.scale

enerzyme.models.cutoff.scale(cutoff_fn: Callable[[Tensor, Tensor, Tensor], Tensor])