in sparse_autoencoder/kernels.py [0:0]
def triton_coo_sparse_dense_matmul( coo_indices: torch.Tensor, coo_values: torch.Tensor, dense: torch.Tensor, N: int, BLOCK_SIZE_AK=128,