flax

48464064 - Simplify _normalize_values and merge test cases

Simplify _normalize_values and merge test cases Use repr(x) as default fallback in _normalize_values to handle all non-serializable objects uniformly. Merge related test cases into comprehensive test_tabulate_with_variable_hooks.