jax_privacy.matrix_factorization.streaming_matrix

Definition of streamin matrix interface.

Functions

diagonal(diag)	An implicit representation of a diagonal matrix.
identity()	An implicit representation of the identity matrix.
momentum_sgd_matrix([momentum, learning_rates])	An implicit representation of the momentum sgd matrix.
multiply_array(A, x)	Computes the matrix-vector product A x.
multiply_streaming_matrices(A, B)	Multiply a StreamingMatrix by another StreamingMatrix.
prefix_sum()	An implicit representation of the lower triangular matrix of ones.
scale_rows_and_columns(matrix[, row_scale, ...])	Returns a new StreamingMatrix with scaled rows and/or cols.

Classes

StreamingMatrix(init_multiply, multiply_next)

A linear mapping x -> A x for a lower-triangular (streaming) A matrix.