Flux.jl
Flux.jl copied to clipboard
feat: Distributed data parallel training support
Support for distributed data parallel training. Inspired by https://github.com/LuxDL/Lux.jl/pull/500 This PR is still work in progress. PR checklist to be continued.
PR Checklist
- [ ] Tests are added
- [ ] Entry in NEWS.md
- [ ] Documentation, if applicable