Rename SN(P)E -> N(P)E in all documentation

Author: michaeldeistler

docs/docs/faq/question_01_leakage.md

@@ -1,22 +1,40 @@
 # What should I do when my 'posterior samples are outside the prior support' in SNPE?
 
-When working with **multi-round** SNPE, you might have experienced the following
-warning:
+When working with **multi-round** NPE (i.e., SNPE), you might have experienced the 
+following warning:
 
 ```python
 Only x% posterior samples are within the prior support. It may take a long time to
 collect the remaining 10000 samples. Consider interrupting (Ctrl-C) and switching to
 'sample_with_mcmc=True'.
 ```
 
-The reason for this issue is described in more detail
-[here](https://arxiv.org/abs/2002.03712) and
+The reason for this issue is described in more detail 
+[here](https://arxiv.org/abs/2210.04815),
+[here](https://arxiv.org/abs/2002.03712), and
 [here](https://arxiv.org/abs/1905.07488). The following fixes are possible:
 
+- use truncated proposals for SNPE (TSNPE)
+```python
+from sbi.inference import NPE
+from sbi.utils import RestrictedPrior, get_density_thresholder
+
+inference = NPE(prior)
+proposal = prior
+for _ in range(num_rounds):
+    theta = proposal.sample((num_sims,))
+    x = simulator(theta)
+    _ = inference.append_simulations(theta, x).train(force_first_round_loss=True)
+    posterior = inference.build_posterior().set_default_x(x_o)
+
+    accept_reject_fn = get_density_thresholder(posterior, quantile=1e-4)
+    proposal = RestrictedPrior(prior, accept_reject_fn, sample_with="rejection")
+```
+
 - sample with MCMC: `samples = posterior((num_samples,), x=x_o, sample_with_mcmc=True)`.
 This approach will make sampling slower, but samples will not "leak".
 
-- resort to single-round SNPE and (if necessary) increase your simulation budget.
+- resort to single-round NPE and (if necessary) increase your simulation budget.
 
 - if your prior is either Gaussian (torch.distributions.MultivariateNormal) or
 Uniform (sbi.utils.BoxUniform), you can avoid leakage by using a mixture density
@@ -25,5 +43,5 @@ interface](https://sbi-dev.github.io/sbi/tutorial/03_flexible_interface/), set
 `density_estimator='mdn'`. When running inference, there should be a print
 statement "Using SNPE-C with non-atomic loss".
 
-- use a different algorithm, e.g., SNRE and SNLE. Note, however, that these algorithms
-can have different issues and potential pitfalls.
+- use a different algorithm, e.g., Sequential NRE and Sequential NLE. Note, however, 
+that these algorithms can have different issues and potential pitfalls.

docs/docs/faq/question_02_nans.md

@@ -16,15 +16,15 @@ Importantly, in order for neural network training work well, the floating point
 number should still be in a reasonable range, i.e., maybe a few standard
 deviations outside of 'good' values.
 
-If you are running **multi-round** SNPE, however, things can go fully wrong if
+If you are running **multi-round** NPE (SNPE), however, things can go fully wrong if
 invalid data are encountered. In that case, you will get the following warning
 
 ```python
 When invalid simulations are excluded, multi-round SNPE-C can leak into the regions
 where parameters led to invalid simulations. This can lead to poor results.
 ```
 
-Hence, if you are running multi-round SNPE and a significant fraction of
+Hence, if you are running multi-round NPE and a significant fraction of
 simulations returns at least one invalid number, we strongly recommend manually
 replacing the value in your simulation code as described above (or resorting to
-single-round SNPE, or using a different `sbi` method entirely).
+single-round NPE, or using a different `sbi` method entirely).

docs/docs/faq/question_04_gpu.md

@@ -10,7 +10,7 @@ Yes, we support GPU training. When creating the inference object in the flexible
 interface, you can pass the `device` as an argument, e.g.,
 
 ```python
-inference = SNPE(prior, device="cuda", density_estimator="maf")
+inference = NPE(prior, device="cuda", density_estimator="maf")
 ```
 
 The device is set to `"cpu"` by default. But it can be set to anything, as long
@@ -33,7 +33,7 @@ device="cuda:0"), covariance_matrix=torch.eye(2, device="cuda:0"))
 
 Whether or not you reduce your training time when training on a GPU depends on
 the problem at hand. We provide a couple of default density estimators for
-`SNPE`, `SNLE` and `SNRE`, e.g., a mixture density network
+`NPE`, `NLE` and `NRE`, e.g., a mixture density network
 (`density_estimator="mdn"`) or a Masked Autoregressive Flow
 (`density_estimator="maf"`). For these default density estimators, we do **not**
 expect a speed-up. This is because the underlying neural networks are relatively

docs/docs/faq/question_06_resume_training.md

@@ -8,7 +8,7 @@ simulations are finished, `sbi` trains a neural network. If this process takes
 too long, you can stop training and resume it later. The syntax is:
 
 ```python
-inference = SNPE(prior=prior)
+inference = NPE(prior=prior)
 inference = inference.append_simulations(theta, x)
 inference.train(max_num_epochs=300)  # Pick `max_num_epochs` such that it does not exceed the runtime.
 

docs/docs/faq/question_07_custom_prior.md

@@ -81,7 +81,7 @@ Note that in `custom_prior_wrapper_kwargs` you can pass additinal arguments for
 the wrapper, e.g., `validate_args` or `arg_constraints` see the `Distribution`
 documentation for more details.
 
-If you are using `sbi` < 0.17.2 and use `SNLE` the code above will produce a
+If you are using `sbi` < 0.17.2 and use `NLE` the code above will produce a
 `NotImplementedError` (see [#581](https://github.com/mackelab/sbi/issues/581)).
-In this case, you need to update to a newer version of `sbi` or use `SNPE`
+In this case, you need to update to a newer version of `sbi` or use `NPE`
 instead.

docs/docs/index.md

@@ -5,7 +5,7 @@ interface:
 
 ```python
 import torch
-from sbi.inference import SNPE
+from sbi.inference import NPE
 
 # define shifted Gaussian simulator.
 def simulator(θ): return θ + torch.randn_like(θ)
@@ -15,7 +15,7 @@ def simulator(θ): return θ + torch.randn_like(θ)
 x = simulator(θ)
 
 # choose sbi method and train
-inference = SNPE()
+inference = NPE()
 inference.append_simulations(θ, x).train()
 
 # do inference given observed data
@@ -103,8 +103,8 @@ The methods then proceed by
    full space of parameters consistent with the data and the prior, i.e. the
    posterior distribution. The posterior assigns high probability to parameters
    that are consistent with both the data and the prior, and low probability to
-   inconsistent parameters. While SNPE directly learns the posterior
-   distribution, SNLE and SNRE need an extra MCMC sampling step to construct a
+   inconsistent parameters. While NPE directly learns the posterior
+   distribution, NLE and NRE need an extra MCMC sampling step to construct a
    posterior.
 4. If needed, an initial estimate of the posterior can be used to adaptively
    generate additional informative simulations.

docs/docs/reference/inference.md

@@ -1,48 +1,48 @@
 # Inference
 
-## Algorithms
+## Training algorithms
 
-::: sbi.inference.snpe.snpe_a.SNPE_A
+::: sbi.inference.trainers.npe.npe_a.NPE_A
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.snpe.snpe_c.SNPE_C
+::: sbi.inference.trainers.npe.npe_c.NPE_C
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.fmpe.fmpe_base.FMPE
+::: sbi.inference.trainers.fmpe.fmpe.FMPE
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.npse.npse.NPSE
+::: sbi.inference.trainers.npse.npse.NPSE
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.snle.snle_a.SNLE_A
+::: sbi.inference.trainers.nle.nle_a.NLE_A
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.snre.snre_a.SNRE_A
+::: sbi.inference.trainers.nre.nre_a.NRE_A
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.snre.snre_b.SNRE_B
+::: sbi.inference.trainers.nre.nre_b.NRE_B
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.snre.snre_c.SNRE_C
+::: sbi.inference.trainers.nre.nre_c.NRE_C
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
 
-::: sbi.inference.snre.bnre.BNRE
+::: sbi.inference.trainers.nre.bnre.BNRE
     selection:
       filters: [ "!^_", "^__", "!^__class__" ]
       inherited_members: true
@@ -59,9 +59,9 @@
 
 ## Helpers
 
-::: sbi.inference.base.infer
+::: sbi.inference.trainers.base.infer
 
-::: sbi.inference.base.simulate_for_sbi
+::: sbi.inference.trainers.base.simulate_for_sbi
 
 ::: sbi.utils.user_input_checks.process_prior