Jax: Actually return derivative and not tuple of derivative

15 jobs for jax_fixes in 20 minutes and 59 seconds (queued for 1 second)
Status Name Job ID Coverage
  Static Checks
passed check_no_asserts #1381353

00:00:14

 
  Build Docker
passed build_docker_from_cache #1381354

00:00:14

 
  Test
passed test_mpi #1381356

00:00:32

passed test_serial #1381355

00:09:25

86.0%
 
  Demo Runs
passed run_bernoulli #1381364

00:00:30

passed run_curve_fitting #1381365

00:00:29

passed run_getting_density #1381363

00:08:44

passed run_getting_started_1 #1381359

00:00:35

passed run_getting_started_2 #1381360

00:00:55

passed run_getting_started_3 #1381361

00:06:00

passed run_getting_started_mf #1381362

00:07:55

passed run_ipynb0 #1381357

00:01:26

passed run_ipynb1 #1381358

00:01:35

passed run_meanfield #1381367

00:01:42

passed run_visual_vi #1381366

00:11:02