v0.8.7
Enhancements
- Torch improvements:
- Added torch implementation for
tracewith axes specified plus tests. - Allowed torch
default_rngseed to be None. - Added axis kwarg support for torch
count_nonzero.
- Added torch implementation for
- Added decorator support for registration APIs:
register_functioncan now be used as a decorator.register_custom_wrappercan now be used as a decorator.
- Refactored internals to use the new decorator style for function registration.