def sample(self, rng: Generator, shape: Optional[Shape]=None) -> Array:
    raise NotImplementedError
def sample(self, rng: Generator, shape: Optional[Shape]=None) -> Array:
    raise NotImplementedError