revert changes + update docstring

15 jobs for fixup-JaxLinearOperator in 21 minutes and 10 seconds (queued for 2 seconds)
Status Job ID Name Coverage
  Static Checks
passed #1378627
check_no_asserts

00:00:14

 
  Build Docker
passed #1378628
build_docker_from_cache

00:00:14

 
  Test
passed #1378630
test_mpi

00:00:32

passed #1378629
test_serial

00:09:31

86.0%
 
  Demo Runs
passed #1378638
run_bernoulli

00:00:32

passed #1378639
run_curve_fitting

00:00:31

passed #1378637
run_getting_density

00:08:43

passed #1378633
run_getting_started_1

00:00:36

passed #1378634
run_getting_started_2

00:00:55

passed #1378635
run_getting_started_3

00:06:02

passed #1378636
run_getting_started_mf

00:07:58

passed #1378631
run_ipynb0

00:01:28

passed #1378632
run_ipynb1

00:01:35

passed #1378641
run_meanfield

00:01:43

passed #1378640
run_visual_vi

00:11:07