gitextract_xp2vjhqo/ ├── .travis.yml ├── AUTHORS ├── LICENSE ├── README.md ├── docs/ │ ├── applications.md │ ├── contact.md │ ├── examples/ │ │ ├── auto_damp.md │ │ ├── convolutional.md │ │ ├── distributed_training.md │ │ └── parameters.md │ ├── index.md │ ├── papers.md │ └── sitemap.md ├── kfac/ │ ├── __init__.py │ ├── examples/ │ │ ├── __init__.py │ │ ├── autoencoder_mnist.py │ │ ├── autoencoder_mnist_tpu_estimator.py │ │ ├── autoencoder_mnist_tpu_strategy.py │ │ ├── classifier_mnist.py │ │ ├── classifier_mnist_tpu_estimator.py │ │ ├── convnet.py │ │ ├── keras/ │ │ │ ├── KFAC_vs_Adam_Experiment.md │ │ │ ├── KFAC_vs_Adam_on_CIFAR10.ipynb │ │ │ └── KFAC_vs_Adam_on_CIFAR10_TPU.ipynb │ │ ├── mnist.py │ │ └── rnn_mnist.py │ └── python/ │ ├── __init__.py │ ├── keras/ │ │ ├── README.md │ │ ├── __init__.py │ │ ├── callbacks.py │ │ ├── optimizers.py │ │ ├── saving_utils.py │ │ └── utils.py │ ├── kernel_tests/ │ │ ├── data_reader_test.py │ │ ├── estimator_test.py │ │ ├── graph_search_test.py │ │ ├── keras_callbacks_test.py │ │ ├── keras_optimizers_test.py │ │ ├── keras_saving_utils_test.py │ │ ├── keras_utils_test.py │ │ ├── layer_collection_test.py │ │ ├── loss_functions_test.py │ │ ├── op_queue_test.py │ │ ├── optimizer_test.py │ │ ├── periodic_inv_cov_update_kfac_opt_test.py │ │ └── utils_test.py │ └── ops/ │ ├── __init__.py │ ├── curvature_matrix_vector_products.py │ ├── estimator.py │ ├── fisher_blocks.py │ ├── fisher_factors.py │ ├── kfac_utils/ │ │ ├── __init__.py │ │ ├── async_inv_cov_update_kfac_opt.py │ │ ├── data_reader.py │ │ ├── data_reader_alt.py │ │ └── periodic_inv_cov_update_kfac_opt.py │ ├── layer_collection.py │ ├── linear_operator.py │ ├── loss_functions.py │ ├── op_queue.py │ ├── optimizer.py │ ├── placement.py │ ├── tensormatch/ │ │ ├── __init__.py │ │ ├── graph_matcher.py │ │ ├── graph_patterns.py │ │ ├── graph_search.py │ │ └── tensorflow_graph_util.py │ └── utils.py └── setup.py