revert changes + update docstring

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

00:00:14

 
  Build Docker
passed build_docker_from_cache #1378628

00:00:14

 
  Test
passed test_mpi #1378630

00:00:32

passed test_serial #1378629

00:09:31

86.0%
 
  Demo Runs
passed run_bernoulli #1378638

00:00:32

passed run_curve_fitting #1378639

00:00:31

passed run_getting_density #1378637

00:08:43

passed run_getting_started_1 #1378633

00:00:36

passed run_getting_started_2 #1378634

00:00:55

passed run_getting_started_3 #1378635

00:06:02

passed run_getting_started_mf #1378636

00:07:58

passed run_ipynb0 #1378631

00:01:28

passed run_ipynb1 #1378632

00:01:35

passed run_meanfield #1378641

00:01:43

passed run_visual_vi #1378640

00:11:07