flax

fa119fda - Remove `ocdbt_merge` and `restore_with_serialized_types` options from `PyTreeCheckpointHandler` as they are not in use anywhere.