class_docstr: Vectorize module takes a function or a module and compiles it for running in parallel on a single device.
class_docstr_tok: ["Vectorize","module","takes","a","function","or","a","module","and","compiles","it","for","running","in","parallel","on","a","single","device","."]
code: def __repr__(self):
      return f"{class_name(self)}(f={self.__wrapped__}, batch_axis={self.batch_axis})"
  
imports: from collections import namedtuple
  from typing import Optional, List, Union, Callable, Tuple
  import jax
  import jax.numpy
  import numpy
  from objax.typing import JaxArray
  from objax.util import class_name, override_args_kwargs, positional_args_names, repr_function
  from objax.variable import BaseVar, RandomState, VarCollection
label_desc: Use image node_id 3 for calling the Vectorize obj's underlying member method code with example usage: obj.__repr__() and returns: class_name, str, self, str, self, str