We are not taking code contributions from Github at this time. All PRs from Github will be rejected. If you find a bug, or for any feature requests and ideas, please email us at kfac-jax-dev@google.com, or raise an issue on the Github issue tracker.
- Issue tracker: https://github.com/deepmind/kfac-jax/issues
- Source code: https://github.com/deepmind/kfac-jax/tree/main