jax

66f52689 - Reenable some test cases that were disabled due to bugs that now seem fixed.