Hyperparam callable improvements

[gpauloski]

Description

Hyperparam callables now take the current K-FAC step as input.
Added reference hyperparameter schedules.
Currently the only one is the exponentially decaying factor running average weight of Martens 2015.

Fixes N/A

Type of Change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• Documentation update (no changes to the code)
• CI change (changes to CI workflows, packages, templates, etc.)

Testing

Update existing unit tests and added new ones accordingly.

Pull Request Checklist

Please confirm the PR meets the following requirements.

• Code changes pass pre-commit (e.g., black, flake8, mypy, etc.).
• Tests have been added to show the fix is effective or that the new feature works.
• New and existing unit tests pass locally with the changes.
• Docs have been updated and reviewed if relevant.