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 Job ID Name Coverage
  Static Checks
passed #1381353
check_no_asserts

00:00:14

 
  Build Docker
passed #1381354
build_docker_from_cache

00:00:14

 
  Test
passed #1381356
test_mpi

00:00:32

passed #1381355
test_serial

00:09:25

86.0%
 
  Demo Runs
passed #1381364
run_bernoulli

00:00:30

passed #1381365
run_curve_fitting

00:00:29

passed #1381363
run_getting_density

00:08:44

passed #1381359
run_getting_started_1

00:00:35

passed #1381360
run_getting_started_2

00:00:55

passed #1381361
run_getting_started_3

00:06:00

passed #1381362
run_getting_started_mf

00:07:55

passed #1381357
run_ipynb0

00:01:26

passed #1381358
run_ipynb1

00:01:35

passed #1381367
run_meanfield

00:01:42

passed #1381366
run_visual_vi

00:11:02