K-FAC: Extensions, improvements, and applications