CQCL · gamatos · Oct 27, 2023 · Oct 27, 2023
diff --git a/docs/densitytensor.rst b/docs/densitytensor.rst
@@ -1,7 +1,8 @@
-densitytensor
+Mixed state simulation
 =======================
 
 .. toctree::
+    :titlesonly:
 
     densitytensor/all_zeros_densitytensor
     densitytensor/kraus

diff --git a/docs/densitytensor/all_zeros_densitytensor.rst b/docs/densitytensor/all_zeros_densitytensor.rst
@@ -1 +1,4 @@
+all_zeros_densitytensor
+==============================================
+
 .. autofunction:: qujax.all_zeros_densitytensor
diff --git a/docs/densitytensor/densitytensor_to_measured_densitytensor.rst b/docs/densitytensor/densitytensor_to_measured_densitytensor.rst
@@ -1 +1,4 @@
+densitytensor_to_measured_densitytensor
+==============================================
+
 .. autofunction:: qujax.densitytensor_to_measured_densitytensor
diff --git a/docs/densitytensor/densitytensor_to_measurement_probabilities.rst b/docs/densitytensor/densitytensor_to_measurement_probabilities.rst
@@ -1 +1,4 @@
+densitytensor_to_measurement_probabilities
+==============================================
+
 .. autofunction:: qujax.densitytensor_to_measurement_probabilities
diff --git a/docs/densitytensor/get_densitytensor_to_expectation_func.rst b/docs/densitytensor/get_densitytensor_to_expectation_func.rst
@@ -1 +1,4 @@
+get_densitytensor_to_expectation_func
+==============================================
+
 .. autofunction:: qujax.get_densitytensor_to_expectation_func
diff --git a/docs/densitytensor/get_densitytensor_to_sampled_expectation_func.rst b/docs/densitytensor/get_densitytensor_to_sampled_expectation_func.rst
@@ -1 +1,4 @@
+get_densitytensor_to_sampled_expectation_func
+==============================================
+
 .. autofunction:: qujax.get_densitytensor_to_sampled_expectation_func
diff --git a/docs/densitytensor/get_params_to_densitytensor_func.rst b/docs/densitytensor/get_params_to_densitytensor_func.rst
@@ -1 +1,4 @@
+get_params_to_densitytensor_func
+==============================================
+
 .. autofunction:: qujax.get_params_to_densitytensor_func
diff --git a/docs/densitytensor/kraus.rst b/docs/densitytensor/kraus.rst
@@ -1 +1,4 @@
+kraus
+==============================================
+
 .. autofunction:: qujax.kraus
diff --git a/docs/densitytensor/partial_trace.rst b/docs/densitytensor/partial_trace.rst
@@ -1 +1,4 @@
+partial_trace
+==============================================
+
 .. autofunction:: qujax.partial_trace
diff --git a/docs/densitytensor/statetensor_to_densitytensor.rst b/docs/densitytensor/statetensor_to_densitytensor.rst
@@ -1 +1,4 @@
+statetensor_to_densitytensor
+==============================================
+
 .. autofunction:: qujax.statetensor_to_densitytensor
diff --git a/docs/index.rst b/docs/index.rst
@@ -44,6 +44,7 @@ Contents
 
 .. toctree::
    :caption: API Reference:
+   :titlesonly:
 
     Pure state simulation <statetensor>
     Mixed state simulation <densitytensor>

diff --git a/docs/statetensor.rst b/docs/statetensor.rst
@@ -1,7 +1,8 @@
-statetensor
+Pure state simulation
 =======================
 
 .. toctree::
+    :titlesonly:
 
     statetensor/all_zeros_statetensor
     statetensor/apply_gate

diff --git a/docs/statetensor/all_zeros_statetensor.rst b/docs/statetensor/all_zeros_statetensor.rst
@@ -1 +1,4 @@
+all_zeros_statetensor
+==============================================
+
 .. autofunction:: qujax.all_zeros_statetensor
diff --git a/docs/statetensor/apply_gate.rst b/docs/statetensor/apply_gate.rst
@@ -1 +1,4 @@
+apply_gate
+==============================================
+
 .. autofunction:: qujax.apply_gate
diff --git a/docs/statetensor/get_params_to_statetensor_func.rst b/docs/statetensor/get_params_to_statetensor_func.rst
@@ -1 +1,4 @@
+get_params_to_statetensor_func
+==============================================
+
 .. autofunction:: qujax.get_params_to_statetensor_func
diff --git a/docs/statetensor/get_params_to_unitarytensor_func.rst b/docs/statetensor/get_params_to_unitarytensor_func.rst
@@ -1 +1,4 @@
+get_params_to_unitarytensor_func
+==============================================
+
 .. autofunction:: qujax.get_params_to_unitarytensor_func
diff --git a/docs/statetensor/get_statetensor_to_expectation_func.rst b/docs/statetensor/get_statetensor_to_expectation_func.rst
@@ -1 +1,4 @@
+get_statetensor_to_expectation_func
+==============================================
+
 .. autofunction:: qujax.get_statetensor_to_expectation_func
diff --git a/docs/statetensor/get_statetensor_to_sampled_expectation_func.rst b/docs/statetensor/get_statetensor_to_sampled_expectation_func.rst
@@ -1 +1,4 @@
+get_statetensor_to_sampled_expectation_func
+==============================================
+
 .. autofunction:: qujax.get_statetensor_to_sampled_expectation_func
diff --git a/docs/utils.rst b/docs/utils.rst
@@ -1,7 +1,8 @@
-utils
+Utility functions
 =======================
 
 .. toctree::
+    :titlesonly:
 
     utils/bitstrings_to_integers
     utils/check_circuit
@@ -10,4 +11,3 @@ utils
     utils/repeat_circuit
     utils/sample_bitstrings
     utils/sample_integers
-
diff --git a/docs/utils/bitstrings_to_integers.rst b/docs/utils/bitstrings_to_integers.rst
@@ -1 +1,4 @@
+bitstrings_to_integers
+==============================================
+
 .. autofunction:: qujax.bitstrings_to_integers
diff --git a/docs/utils/check_circuit.rst b/docs/utils/check_circuit.rst
@@ -1 +1,4 @@
+check_circuit
+==============================================
+
 .. autofunction:: qujax.check_circuit
diff --git a/docs/utils/integers_to_bitstrings.rst b/docs/utils/integers_to_bitstrings.rst
@@ -1 +1,4 @@
+integers_to_bitstrings
+==============================================
+
 .. autofunction:: qujax.integers_to_bitstrings
diff --git a/docs/utils/print_circuit.rst b/docs/utils/print_circuit.rst
@@ -1 +1,4 @@
+print_circuit
+==============================================
+
 .. autofunction:: qujax.print_circuit
diff --git a/docs/utils/repeat_circuit.rst b/docs/utils/repeat_circuit.rst
@@ -1 +1,4 @@
+repeat_circuit
+==============================================
+
 .. autofunction:: qujax.repeat_circuit
diff --git a/docs/utils/sample_bitstrings.rst b/docs/utils/sample_bitstrings.rst
@@ -1 +1,4 @@
+sample_bitstrings
+==============================================
+
 .. autofunction:: qujax.sample_bitstrings
diff --git a/docs/utils/sample_integers.rst b/docs/utils/sample_integers.rst
@@ -1 +1,4 @@
+sample_integers
+==============================================
+
 .. autofunction:: qujax.sample_integers