code: def generate_quantization_constants(alpha, beta, alpha_q, beta_q):
      # Affine quantization mapping
      s = (beta - alpha) / (beta_q - alpha_q)
      z = int((beta * alpha_q - alpha * beta_q) / (beta - alpha))
  
      return s, z
  
imports: import typing
  from functools import partial
  import jax
  import jax.numpy
  import matplotlib.pyplot
  import numpy
  import optax
  from datasets import load_dataset
  from flax.experimental import nnx
label_desc: Use image node_id 15 for calling a global function with example usage: generate_quantization_constants(alpha, beta, alpha_q, beta_q) and returns: s,  z