kfac-jax 0.0.5.

botev released this 16 May 18:03

· 236 commits to main since this release

aa73ceb

What's Changed

- Modifying the structure of the experiments class by separating the Supervised experiment from the Jaxline experiment inheritance. by @copybara-service in #122
- Introducing a more general KroneckerFactored block class. by @copybara-service in #123
- Bumping up the package version due to changes in the curvature block classes. by @copybara-service in #124

Full Changelog: v0.0.4...v0.0.5

Assets 2