`haliax.debug.diagnose_common_issues`

[dlwh]

hax.debug.diagnose_common_issues is a function that will raise an exception if it detects problems with your module. Currently, we diagnose:

• Reuse of arrays or NamedArrays in a field. Equinox modules must be trees.
• Use of arrays or NamedArrays in a static field. Static data in JAX/Equinox must be hashable, and arrays are not hashable.

I'll probably call this by default in Levanter