flax
6226a9cc - Enable support for non-string keys in state dicts.

Commit

3 years ago

Enable support for non-string keys in state dicts. This implies Modules can have a dictionary assignment with non-stirng keys.

References

Author

jheek

Parents