Fix docs

Author: penelopeysm

docs/src/api.md

@@ -123,15 +123,6 @@ The `predict` function has two main methods:
 predict
 ```
 
-## Marginalization
-
-DynamicPPL provides the `marginalize` function to marginalize out variables from a model.
-This requires `MarginalLogDensities.jl` to be loaded in your environment.
-
-```@docs
-marginalize
-```
-
 ### Basic Usage
 
 The typical workflow for posterior prediction involves:
@@ -145,6 +136,15 @@ When using `predict` with `MCMCChains.Chains`, you can control which variables a
   - `include_all=false` (default): Include only newly predicted variables
   - `include_all=true`: Include both parameters from the original chain and predicted variables
 
+## Marginalization
+
+DynamicPPL provides the `marginalize` function to marginalize out variables from a model.
+This requires `MarginalLogDensities.jl` to be loaded in your environment.
+
+```@docs
+marginalize
+```
+
 ## Models within models
 
 One can include models and call another model inside the model function with `left ~ to_submodel(model)`.