Optimisers.jl
Optimisers.jl copied to clipboard
Add `setup(::Function, model)`
Quick sketch of one way to easily allow different rules for different arrays, by modifying setup -- see docstring.
PR Checklist
- [ ] Tests are added
- [ ] Documentation, if applicable