code: def out_qdq_bwd(compute_dtype, res, g):
      scale, amax_history = res
      q_g, new_scale, new_history = qdq_and_return(
          g, jnp.float8_e5m2, scale, amax_history, compute_dtype
      )
      return q_g, new_scale, new_history
  
imports: import warnings
  from functools import partial
  from jax import custom_jvp, custom_vjp, lax, random
  from jax import numpy
  from flax.linen import initializers, module
label_desc: Use image node_id 13 for calling a global function with example usage: out_qdq_bwd(compute_dtype, res, g) and returns: q_g,  new_scale,  new_history