Skip to content
Advertisement

Tag: tensorflow-probability

How to run scipy’s BFGS on GPU

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

Advertisement