Class CliArgs

src/fused_mm_sampling/bench/speed_test.py:27–29

class CliArgs(Args, cli_parse_args=True):
    case: str = "small"
    n_hidden_states: int = 1


sample_compiled = torch.compile(sample)