Simulation-based inference with the Python Package sbijax
Neural simulation-based inference (SBI) describes an emerging family of methods for Bayesian inference with intractable likelihood functions that use neural networks as surrogate models. Here we introduce sbijax, a Python package that implements a wide variety of state-of-the-art methods in neural simulation-based inference using a user-friendly programming interface. sbijax offers high-level functionality to quickly construct SBI estimators, and compute and visualize posterior distributions with only a few lines of code. In addition, the package provides functionality for conventional approximate Bayesian computation, to compute model diagnostics, and to automatically estimate summary statistics. By virtue of being entirely written in JAX, sbijax is extremely computationally efficient, allowing rapid training of neural networks and executing code automatically in parallel on both CPU and GPU.
Code (0)
등록된 구현이 없습니다.
Tasks
Bayesian InferenceCPUGPUSimilar Papers 제목 키워드 기반
CausalML: Python Package for Causal Machine Learning
CausalML is a Python implementation of algorithms related to causal inference and machine learning. Algorithms combining causal inference and machine learning have been a trending topic in recent years. This package trie…
BIG-bench Machine LearningCausal Inferencepgmpy: A Python Toolkit for Bayesian Networks
Bayesian Networks (BNs) are used in various fields for modeling, prediction, and decision making. pgmpy is a python package that provides a collection of algorithms and tools to work with BNs and related models. It imple…
Causal DiscoveryCausal IdentificationCausal InferenceDecision Making+1Using Python for Model Inference in Deep Learning
Python has become the de-facto language for training deep neural networks, coupling a large suite of scientific computing libraries with efficient libraries for tensor computation such as PyTorch or TensorFlow. However, …
Deep LearningmodelModel extractionGraSPy: Graph Statistics in Python
We introduce GraSPy, a Python library devoted to statistical inference, machine learning, and visualization of random graphs and graph populations. This package provides flexible and easy-to-use algorithms for analyzing …
BIG-bench Machine Learningsi4onnx: A Python package for Selective Inference in Deep Learning Models
In this paper, we introduce si4onnx, a package for performing selective inference on deep learning models. Techniques such as CAM in XAI and reconstruction-based anomaly detection using VAE can be interpreted as methods …
Anomaly DetectionDeep Learning