in sparse_autoencoder/train.py [0:0]
def __init__( self, n_dirs_local: int, d_model: int, k: int, auxk: int | None, dead_steps_threshold: int, comms: ShardingComms | None = None,