kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

232 stars

11 watching

18 forks

Language: Python

last commit: 4 days ago

Linked from 1 awesome list

bayesian-deep-learningmachine-learningoptimization

Backlinks from these awesome lists:

n2cholas/awesome-jax