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.
