kfac_jax/__init__.py,sha256=TnMiUENOBTpmRMMoHvs-nFVbVZcHxFlPBnA5_gMZ9HA,7768
kfac_jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
kfac_jax/_src/__init__.py,sha256=WIVmD14Sr1HfmJddHhkHkq6FPL1k3Dpk09EDS0ET13M,615
kfac_jax/_src/layers_and_loss_tags.py,sha256=8INWjlePwnOnDiWHfQuW-KMSfDq1kVI2yCu3OXKbD50,15033
kfac_jax/_src/loss_functions.py,sha256=5eoVAxcKcKB-ZYPcy-k7PBUEAvPZZeejpb0nsqLgh4U,49248
kfac_jax/_src/optimizer.py,sha256=zZWBnrq_Ab-WRmaziCG-_-rEFXhI8VgggavJh7TJiec,79318
kfac_jax/_src/patches_second_moment.py,sha256=yTEmMJ6q4IBTP96gozlOdolvSlQaBRptA-ugkRt-Cjw,36845
kfac_jax/_src/tag_graph_matcher.py,sha256=gLOvnY6SjzfQ3zWQMi9UBA1oAqfX0NyyoEUMMYOJ670,70488
kfac_jax/_src/tracer.py,sha256=a704oo2b67RG-JmwtqpbTD2gdo5jEmy_frB2H9jTkA0,43520
kfac_jax/_src/curvature_blocks/__init__.py,sha256=az0VQ-6sB_1nL1ELTPaYSvGbZlVRPAf7yFA2eylxE9Y,2209
kfac_jax/_src/curvature_blocks/curvature_block.py,sha256=IPjQEhdoE87Jb4n3nKBuw1sN8vZV3c4zXSwxet4eynM,20181
kfac_jax/_src/curvature_blocks/diagonal.py,sha256=LPhm9Rc-wganJ_twnI_2Bcd9azjMSTC0JRZz8ZIHNYw,11417
kfac_jax/_src/curvature_blocks/full.py,sha256=bqu9QLngPzp1aoJCeSNYiPicPepAJq4lB9cSLyPzreQ,17067
kfac_jax/_src/curvature_blocks/kronecker_factored.py,sha256=GggvVHqyEmPms-ttX79cjNT2saMdvZLoCUZeydyfpXY,22469
kfac_jax/_src/curvature_blocks/tnt.py,sha256=zW5uDxpnm8XRNe3CyWa9QpX_2fEJm9y1EsaEUB39dEU,9071
kfac_jax/_src/curvature_blocks/utils.py,sha256=Gw6245bTqlNNMW8cKiov15QPkVu7UzcYrRbQ4oP7jGE,4836
kfac_jax/_src/curvature_estimator/__init__.py,sha256=1YKWFoMudZL8Eh05KPhkDgc1ewms0psYqGJLFmsm9Z8,4066
kfac_jax/_src/curvature_estimator/block_diagonal.py,sha256=ePRCg9lakLb73rHUozBwSndHosX4Em3S0Q9Ny6_JRMk,35701
kfac_jax/_src/curvature_estimator/curvature_estimator.py,sha256=yK9gHmqDjtONO6EukEPjK6mJ0SUk7U9SKVvMtcxWVQc,14415
kfac_jax/_src/curvature_estimator/explicit_exact.py,sha256=MTbM-IgGIHjAXkYGO8S_52WpxjTMU9S8WX24YiG_ITA,6366
kfac_jax/_src/curvature_estimator/implicit_exact.py,sha256=xSCPuqFY6a3HQQi90z22SKVOVAYAxRqSEEF4N9Eemkg,17162
kfac_jax/_src/curvature_estimator/optax_interface.py,sha256=XnzFJozftZ5t-X4Ec8g046mC4ojPwgMpIZwxQKV8WMg,19230
kfac_jax/_src/utils/__init__.py,sha256=NLKnBP6nq2OTCMadjdZYGB_dQ0uSQ0O4LjnZ2j5CDhc,5537
kfac_jax/_src/utils/accumulators.py,sha256=Gk1XjftIuvo_00fwyIikiPcMH7STNe1V_XkJlc_CtVI,9462
kfac_jax/_src/utils/math.py,sha256=jjz9s9vMJUjfu2tMu9SB3Spf3gFKLnoY_fgPNRyquxg,40910
kfac_jax/_src/utils/misc.py,sha256=NkDeCuIQE1MHTT4Xq6fyDdsJLlf0_MzSv9cJUcjWOew,13204
kfac_jax/_src/utils/parallel.py,sha256=zJZ_q0JgOoyv7Zc_H8AITGH3IWoecyrrcLk-1uqKe-0,12766
kfac_jax/_src/utils/staging.py,sha256=vmZzDqX5ZFoAnkk6P52VnDmDa1OxRCxPRqL5AJpBK5g,10062
kfac_jax/_src/utils/types.py,sha256=UY3GRE7RTNA-SzC2L0CLDhnUKUG95iuQjQMsi-XgJ3U,2964
kfac_jax-0.0.8.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
kfac_jax-0.0.8.dist-info/METADATA,sha256=X0Jk2NSll9ThPngW0LwaCJLqq1UjAglENzjqfEsXbzE,23597
kfac_jax-0.0.8.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
kfac_jax-0.0.8.dist-info/top_level.txt,sha256=1qH6OpEVVLTL9XkA468A0ZDad088AHPpMovTNh-V8Ns,9
kfac_jax-0.0.8.dist-info/RECORD,,
