SoftCap.Config(
        channels_in=4,
        channels_out=4,
        cap=2.0,
        inner=Linear.Config(
            channels_in=4,
            channels_out=4,
            bias=False,
            depth_index=(),
            device=None,
            dtype=None,
            init_weight=priml.model.init.kaiming_uniform,
            init_bias=torch.nn.init.zeros_,
            linear_fn=<built-in function linear>,
            shard=None
        ),
        dtype=torch.float32
    )
