Better error report for max traces exceeded

[uduse]

In my experience, when chex reports max traces exceeded, it's usually because of me passing parameters to the function with different shapes or data types. Is it possible for chex to report such inconsistency?

e.g.,

AssertionError: [Chex] Function '_wrapper' is traced > 1 times!
Difference in input shapes. Last time variable `x` traced with shape "(10, 1)", this time traced with shape "(9, 1)".

[hbq1]

Hey, thanks for the question - yes, it can be added! Would you like to send a PR?

[uduse]

@hbq1 I actually don't know how to do it in chex. I am doing stupid prints to debug this...