kfac_jax/__init__.py,sha256=pyyJLGSPAOvsL0rFuSVbvIj_p0xMSSS90EH04BKrapM,7672
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=5n-WsDE0zJhDfDLHilGfdYj37M48y4FqucmBvM5qGaw,14957
kfac_jax/_src/loss_functions.py,sha256=5eoVAxcKcKB-ZYPcy-k7PBUEAvPZZeejpb0nsqLgh4U,49248
kfac_jax/_src/optimizer.py,sha256=q2x6-KirTzyoZkZZV_B6PNHBm7RWRIy990xJmZGQVNc,68511
kfac_jax/_src/patches_second_moment.py,sha256=Q4XQmKaFbErY4vP024BPLgUisZx2ZCX9tA_nbRyqvjs,36749
kfac_jax/_src/tag_graph_matcher.py,sha256=Pe1D_xdAV18BNObN7lFF24dtre8TEMthAXhMowrhhuI,68317
kfac_jax/_src/tracer.py,sha256=08asfK-YooCEWbDXHQXiEiprtNeKUViCteL9RB4SviA,41962
kfac_jax/_src/curvature_blocks/__init__.py,sha256=LF7jnh5Zn-kZQz2qs4pZOPgbxzw2hKMLVwGEPpE_xlU,2177
kfac_jax/_src/curvature_blocks/curvature_block.py,sha256=IPjQEhdoE87Jb4n3nKBuw1sN8vZV3c4zXSwxet4eynM,20181
kfac_jax/_src/curvature_blocks/diagonal.py,sha256=DAK6ajgolHUaEuLRv5kOj_gHAhW78kDCtRFpuLfNVVI,11415
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=TjiGMpnUJUKH3L9SVSYg5p1AKqiCTt4IfMT9__uPgZc,9068
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=RAQhunRRLemlISqeU6XonknqAJV42s90UW3ctO3jOwM,30056
kfac_jax/_src/curvature_estimator/curvature_estimator.py,sha256=k6SkD5E0jCeJYbcEz4_m8ABA4U5WLJXJUmyEmxbcUGU,13544
kfac_jax/_src/curvature_estimator/explicit_exact.py,sha256=zPrTe5TmsM8tqAliInEZFinXP51ihWtSxtvBWx2ESWM,5641
kfac_jax/_src/curvature_estimator/implicit_exact.py,sha256=xSCPuqFY6a3HQQi90z22SKVOVAYAxRqSEEF4N9Eemkg,17162
kfac_jax/_src/curvature_estimator/optax_interface.py,sha256=GDHDsYTXZybb2txISx3gwkZcQ_xq0qIr1K_nygBFKxU,19186
kfac_jax/_src/utils/__init__.py,sha256=FlfChiuIndg1hzn2MDRrWyJDEkag_v32Oo9QSWRD8Ls,5468
kfac_jax/_src/utils/accumulators.py,sha256=Gk1XjftIuvo_00fwyIikiPcMH7STNe1V_XkJlc_CtVI,9462
kfac_jax/_src/utils/math.py,sha256=CHSl8VHVIKLI4K-VvR0bZmD45AGlqnvOv6V3_kancSE,39495
kfac_jax/_src/utils/misc.py,sha256=kTGqsk4A8B0VnSzueT0pz7E9ovCdwLVDASt6UfSSBR4,12393
kfac_jax/_src/utils/parallel.py,sha256=UdmWW79NqV0zY_pNtotkp-eOb2ICwMZU8zMqoetPmKk,9989
kfac_jax/_src/utils/staging.py,sha256=i0YF-WtzGDfJhaKt_RKrTYd6DuzfIVfhAwZewLBTaEM,10191
kfac_jax/_src/utils/types.py,sha256=UY3GRE7RTNA-SzC2L0CLDhnUKUG95iuQjQMsi-XgJ3U,2964
kfac_jax-0.0.7.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
kfac_jax-0.0.7.dist-info/METADATA,sha256=708RWtInIfr0Drw-uUmobXvA4OABX6t3uRNLT-b1Fx8,23597
kfac_jax-0.0.7.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
kfac_jax-0.0.7.dist-info/top_level.txt,sha256=1qH6OpEVVLTL9XkA468A0ZDad088AHPpMovTNh-V8Ns,9
kfac_jax-0.0.7.dist-info/RECORD,,
