GitHub repository: mahdi-shafiei/kfac-jax
Primary language: Python
Description: Second Order Optimization and Curvature Estimation with K-FAC in JAX.