I’d like to run scipy implementation of BFGS optimization algorithm on GPU and scipy seems not to support GPUs. The target function which I want to run on GPU is the following one which is part of the implementation of this repository: I know there is Tensorflow Probablity implementation of BFGS, but I couldn’t find out how I can convert