in sparse_autoencoder/kernels.py [0:0]
def triton_sparse_transpose_dense_matmul( sparse_indices: torch.Tensor, sparse_values: torch.Tensor, dense: torch.Tensor, N: int, BLOCK_SIZE_AK=128,