flax

Added an error for when we call init, init_with_output and apply on Module class.

#2529

Merged