Sebastian Khan
Results
11
comments of
Sebastian Khan
Ah! Great! It's a precision issue. Setting JAX to appropriately ``` from jax.config import config config.update("jax_enable_x64", True) ``` Now with diag_reg=1e-8 I get the following! ![Screenshot 2020-10-12 at 16 05...