jax

bbf8129a - Change test tolerance logic not to choose tolerance values based on f… (#1701)

Change test tolerance logic not to choose tolerance values based on f… (#1701) * Change test tolerance logic not to choose tolerance values based on flags (in particular, --jax_enable_x64). We would like to move away from having global flags to enable 64-bit mode. We therefore need other methods to select test tolerances. Instead, use a per-type default tolerance, and allow tests to pass per-type dictionaries of tolerances as atol and rtol values. Fix up a number of tolerances to make tests pass. * Fix test tolerances. * Fix dtype canonicalization for test tolerances. * Relax core test_vjp tolerance.