Restructure aesaraf docs order

ricardoV94 · ricardoV94 · commit 7684ec6b2ca4 · 2022-09-28T08:49:16.000+02:00
diff --git a/docs/source/api/aesaraf.rst b/docs/source/api/aesaraf.rst
@@ -10,17 +10,16 @@ Aesara utils
    gradient
    hessian
    hessian_diag
+   jacobian
    inputvars
    cont_inputs
    floatX
    intX
    smartfloatX
-   jacobian
    CallableTensor
    join_nonshared_inputs
    make_shared_replacements
    generator
    set_at_rng
    at_rng
-   take_along_axis
    pandas_to_array
diff --git a/pymc/aesaraf.py b/pymc/aesaraf.py
@@ -81,6 +81,7 @@
     "set_at_rng",
     "at_rng",
     "convert_observed_data",
+    "compile_pymc",
 ]
 
 
