deepmind/kfac-jax: Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Category: math
Language: Python
Topics: bayesian-deep-learning, jax, machine-learning, optimization