Wrap PyTorch models under `TorchModule`#147

Merged

bryanlimy merged 9 commits intomainfrom pytorch_wrapper

Feb 9, 2024

+172-74

Commits on Feb 6, 2024

create neural_networks wrapper in emulators where each torch model implement the TorchModule class and user can define which NN architecture to use by providing module argument
bryanlimy
committed
remove unused import
bryanlimy
committed