Creating a NNX model with no bias results in a None being passed to the overview.

PiperOrigin-RevId: 720549126

Author: CLU Authors

clu/parameter_overview.py

@@ -98,6 +98,13 @@ def count_parameters(params: _ParamsContainer) -> int:
 
 
 def _make_row(name, value) -> _ParamRow:
+  if value is None:
+    return _ParamRow(
+        name=name,
+        shape=(),
+        dtype="",
+        size=0,
+    )
   return _ParamRow(
       name=name,
       shape=value.shape,