r"""Validation helper to check that the amount of parameters to prune is meaningful wrt to the size of the data (`tensor_size`). Args:

r"""Validation helper to check the range of amount at init. Args: amount (int or float): quantity of parameters to prune. If f

r"""AlexNet model architecture from the `"One weird trick..." <https://arxiv.org/abs/1404.5997>`_ paper. The required minimum input size of th

r"""The ``build()`` method must be overridden to define neural network computaion logic. The ``build()`` method of nn.Graph is very s

Generates a random sample from a given 1-D array, which aligns with numpy.random.choice see https://numpy.org/doc/stable/reference/random/generate

r"""Applies the latest ``method`` by computing the new partial masks and returning its combination with the ``default_mask``. The new

r"""Returns local rank as device index.

Parses a flat list of parameter tensors into a list of CellParams