jax

f069eade - refactor: replace assert False with proper ValueError in serialization

refactor: replace assert False with proper ValueError in serialization Two 'assert False, kind' statements in serialization.py could be disabled with python -O and provided poor error messages. This changes them to raise ValueError with descriptive messages, matching the existing error handling pattern in the same file. - Line 380: _deserialize_pytreedef_to_pytree - Line 721: _deserialize_disabled_safety_check