In particular, we solve the optimization problem using a gradient-based variational inference algorithm called Stein Variational Gradient Descent. This allows us to find a distribution of solutions ...

Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditioner and more) ...

Dataset analysis, preprocessing, metrics and other analyses are shown in Jupyter notebook binary_classification.ipynb.