guitarists: integrating neural amp modeling python code?

Neural amp modeler (https://www.neuralampmodeler.com/) is a python package that uses audio files to accurately model amplifiers and speaker cabinets. There is a VST that works in Max and a standalone application. Both use profiles and impulse responses that users can upload here: https://www.tone3000.com/

The code is open source and I'm wondering if integrating it (not just the VST) in Max would be doable or advantageous for cpu utilization.

Thoughts?