I tried to run your collab but I get this error trying to use auto_LiRPA :
File "/usr/local/lib/python3.8/dist-packages/auto_LiRPA/operators/base.py", line 8, in <module> from ..perturbations import *
File "/usr/local/lib/python3.8/dist-packages/auto_LiRPA/perturbations.py", line 78 if type(A) != np.ndarray: ^ IndentationError: expected an indented block