Skip to content

Commit dc477e0

Browse files
committed
-
1 parent 0f6f6c9 commit dc477e0

1 file changed

Lines changed: 8 additions & 10 deletions

File tree

paper/paper.md

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,15 @@ data and then numerically constructing an appropriate approximation to this pseu
4343
On the one hand, this family of methods has been shown to be more computationally efficient and often more accurate than ABC methods, in particular for smaller simulation budgets. On the other hand,
4444
SBI allows to easily amortize inference, i.e., to infer the posterior distribution for multiple different observations once a neural model has been trained.
4545

46-
::: {#tbl-methods}
46+
Here we propose `sbijax`, a Python package implementing state-of-the-art methodology of neural simulation-based inference.
47+
While the main focus of the package is the implementation of recent algorithms to make them available to practitioners, e.g., @wildberger2023flow or @schmitt2023consistency,
48+
`sbijax` also implements common methods from approximate Bayesian computation, e.g., SMC-ABC [@beaumont2009adaptive], to have the entire SBI toolbox in one efficient package (see Table 1 for an overview).
49+
In addition, `sbijax` provides functionality for model diagnostics, posterior visualization and Markov Chain Monte Carlo (MCMC) sampling.
50+
The package uses the high-performance computing framework `JAX` as a backend [@jax2018github].
51+
Using `JAX` has several advantages, including a) that it uses the same syntax as `numpy` [@harris2020array] which enables a seamless transition for applied scientists who already are familiar with it,
52+
and b) that empirical evaluations have shown that `JAX` can be significantly faster than `PyTorch` (see, e.g., @phan2019composable).
4753

48-
| Model | Class name | Reference |
54+
| **Model** | **Class name** | **Reference** |
4955
|------------------------------------------------|------------|-----------------------------|
5056
| Sequential Monte Carlo ABC | `SMCABC` | @beaumont2009adaptive |
5157
| Neural likelihood estimation | `NLE` | @papamakarios2019sequential |
@@ -61,14 +67,6 @@ SBI allows to easily amortize inference, i.e., to infer the posterior distributi
6167

6268
:Implemented SBI methods in `sbijax`.
6369

64-
Here we propose `sbijax`, a Python package implementing state-of-the-art methodology of neural simulation-based inference.
65-
While the main focus of the package is the implementation of recent algorithms to make them available to practitioners, e.g., @wildberger2023flow or @schmitt2023consistency,
66-
`sbijax` also implements common methods from approximate Bayesian computation, e.g., SMC-ABC [@beaumont2009adaptive], to have the entire SBI toolbox in one efficient package (see @tbl-methods for an overview).
67-
In addition, `sbijax` provides functionality for model diagnostics, posterior visualization and Markov Chain Monte Carlo (MCMC) sampling.
68-
The package uses the high-performance computing framework `JAX` as a backend [@jax2018github].
69-
Using `JAX` has several advantages, including a) that it uses the same syntax as `numpy` [@harris2020array] which enables a seamless transition for applied scientists who already are familiar with it,
70-
and b) that empirical evaluations have shown that `JAX` can be significantly faster than `PyTorch` (see, e.g., @phan2019composable).
71-
7270
# State of the field
7371

7472
While a plethora of different models has been proposed in the recent literature, the development of adequate software packages has not followed at the same pace,

0 commit comments

Comments
 (0)