trellis_sparse_mlx