Fix embedding field annotation

Commit

4 years ago

Fix embedding field annotation flax params should not be used in hash and comparison checks.