flax

Linen `Module` instances are now Frozen after `setup` has been called.

#823

Merged