flax

68ce7afe - Improve error reporting for Modules that contain un-hashable attributes