code: def testHardTanhMemory(self):
      # see https://github.com/google/jax/pull/1640
      with jax.enable_checks(
          False
      ):  # With checks we materialize the array
          jax.make_jaxpr(lambda: nn.hard_tanh(jnp.ones((10**12,))))
  
imports: import collections
  from functools import partial
  import itertools
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import scipy.stats
  from jax._src import config
  from jax._src import core
  from jax._src import test_util
  from jax._src import ad_checkpoint
  from jax.test_util import check_grads
  from jax import nn
  from jax import random
  import jax
  import jax.numpy
label_desc: Use image node_id 23 for calling the NNFunctionsTest obj's underlying member method code with example usage: obj.testHardTanhMemory() without return types