jax_privacy.matrix_factorization.optimization.jax_enable_x64

jax_privacy.matrix_factorization.optimization.jax_enable_x64(fn)[source]

Decorator to enable x64 precision for a function.