Clean up JAX/NumPy boundary in matrix_factorization: use NumPy for#296

copybara-service[bot] merged 1 commit into

maingoogle-deepmind/jax_privacy:mainfrom

test_938206405google-deepmind/jax_privacy:test_938206405Copy head branch name to clipboard