Open alibool opened 2 years ago
Hi @alibool - thanks for your comment and suggestion.
This would require a significant refactor. I'd have to step back through each minimization routine, add type hints to the function signatures and make sure that all torch ops are TorchScript-supported. It's on my agenda but it won't be a high priority (unless I have some help).
Thanks for this great package, it seems that
minimize
is not jit-able? for exampleWill you consider making
minimize
jit-able?