code: def _bwd_wrapper(treedef, bwd_fn, tangent):
      vars_grad, *inputs_grad = bwd_fn(tangent)
      vars_grad = treedef.unflatten(vars_grad)
      return (vars_grad, *inputs_grad)
  
imports: import collections
  import dataclasses
  import functools
  from typing import Any, Callable, Dict, Generic, Iterable, List, Mapping, Optional, Sequence, Tuple, TypeVar, Union
  import warnings
  from flax import traceback_util
  import jax
  from jax import random
  from .None import axes_scan, meta
  from .frozen_dict import freeze, unfreeze
  from .scope import CollectionFilter, DenyList, Filter, PRNGSequenceFilter, Scope, group_collections, in_filter, intersect_filters, is_filter_empty, subtract_filters, union_filters
label_desc: Use image node_id 10 for calling a global function with example usage: _bwd_wrapper(treedef, bwd_fn, tangent) and returns: vars_grad,  unknown