Skip to content

kfac-jax 0.0.2.
Compare
Choose a tag to compare
@botev botev released this 07 Jun 14:34
· 243 commits to main since this release
v0.0.2
9832b84

What's Changed

  • Moving .github to top-level directory for CI. by @copybara-service in #1
    • Updated documentation for state classes. by @copybara-service in #2
  • Changing the name on PyPi to kfac-jax. by @copybara-service in #3
  • Making the tracer test in float64. by @copybara-service in #4
    • Allowing graph patterns with multiple broadcast to be merged without dangling equations. by @copybara-service in #5
    • Adding README for the examples. by @copybara-service in #7
  • Changing deprecated tree_multimap to tree_map. by @copybara-service in #8
  • Fixing small error introduced due to updates to chex. by @copybara-service in #11
  • Fixing typo "drop_reminder" by @copybara-service in #13
    • Adding an argument to set the reduction ratio thresholds for automatic damping adjustment. by @copybara-service in #12
    • Adding "modifiable_attribute_exceptions" argument to optimizer by @copybara-service in #14
  • Changing Imagenet dataset in examples to use a seed for file shuffling to achieve determinism. by @copybara-service in #17
  • Small fix to a doc reference bug. by @copybara-service in #16
  • Making WeightedMovingAverage to work with arbitrary structures. by @copybara-service in #19
    • Minor typos. by @copybara-service in #20
  • Correct buffer donation of Optimizer._step. by @copybara-service in #21
  • Replacing yield from with direct iteration. by @copybara-service in #24
  • Adding stepwise schedule option to examples. by @copybara-service in #18
  • Publishing a new version to PyPi. by @copybara-service in #28

New Contributors

  • @copybara-service made their first contribution in #1

Full Changelog: https://github.com/deepmind/kfac-jax/commits/v0.0.2

Assets 2
Loading