beta-divergence
A pytorch package for non-negative matrix factorization.
NumPy and PyTorch implementations of the beta-divergence loss.