Version 0.2.1

What's Changed

• Bnb/masked fwp by @bnb32 in #231
• Gb/optm state by @grantbuster in #242
• uv invert for netcdf writes by @bnb32 in #243
• sum multi gpu gradients and apply once to weight equally in adam mome… by @grantbuster in #244
• miscellaneous bias module clean up by @bnb32 in #245
• Gb/disc valid by @grantbuster in #246
• Bnb/1d nc loading by @bnb32 in #247
• Gb/qa fix by @grantbuster in #248

Full Changelog: v0.2.0...v0.2.1