optax.ZeroNansState# class optax.ZeroNansState( found_nan: Any, )[source]# Contains a tree. The entry found_nan has the same tree structure as that of the parameters. Each leaf is a single boolean which contains True iff a NaN was detected in the corresponding parameter array at the last call to update.
