code: def zhugupta_func(progress: float) -> float:
      """From 'To Prune or Not To Prune' :cite:`zhu2017prune`."""
      return 1.0 - (1.0 - progress) ** 3
  
imports: from collections.abc import Iterator, Mapping, Sequence
  import functools
  from typing import Callable
  from absl import app
  import haiku
  import jax
  import jax.numpy
  import numpy
  import optax
  import tensorflow_datasets
label_desc: Use image node_id 2 for calling a global function with example usage: zhugupta_func(progress) and returns: unknown