Weights may need updating for upcoming Flux deprecations

While testing #32, I noticed this triggered in the log: https://github.com/FluxML/Flux.jl/blob/0a215462ad8e0ba795205c9e94864403207d63fa/src/deprecations.jl#L13.

On a side note, should the badges in the README point to GHA now?