Jit scripting

[zaccharieramzi]

Hi !

Has anyone tried to jit script functions implemented using torchopt, and if so did you manage to make it work?

At the moment I am facing an issue similar to:

RuntimeError: 
undefined value torchopt:
    ) -> torch.Tensor:
        optimizer = torchopt.FuncOptimizer(torchopt.adam(lr=self.lr))
                    ~~~~~~~~ <--- HERE
    
        with torch.enable_grad():