Pythonパッケージ sbijax を用いたシミュレーションベースの推論

概要：ニューラルシミュレーションベースの推論（SBI）は、ニューラルネットワークを代理モデルとして用いる、計算が困難な尤度関数を持つベイズ推論の新興手法群を説明します。ここでは、ユーザーフレンドリーなプログラミングインターフェースを用いて、ニューラルシミュレーションベースの推論における最先端手法を幅広く実装した Python パッケージ sbijax を紹介します。sbijax は、SBI 推定量を迅速に構築し、わずか数行のコードで事後分布を計算・可視化するための高レベル機能を提供します。さらに、このパッケージは従来の近似ベイズ推定（ABC）用の機能、モデル診断の算出、および要約統計量を自動的に推定する機能を提供します。完全に JAX で記述されているため、sbijax は非常に計算効率が高く、ニューラルネットワークの迅速な訓練と、CPU および GPU の両方でコードを自動的に並列実行することを可能にします。