Вход на сайт

Просмотр новости

Найдите то, что Вас интересует

py/cuTAGI: An Open-Source Library for Tractable Approximate Gaussian Inference in Bayesian Neural Networks

Дата публикации: 17-08-2026 20:26:00


This paper introduces pyTAGI, a Python wrapper, and cuTAGI, its high-performance C++/CUDA backend, implementing Tractable Approximate Gaussian Inference (TAGI) for neural networks. TAGI treats all network quantities as Gaussian random variables and derives closed-form expressions for prior/posterior expected values, variances, and covariances, enabling analytic Bayesian learning without relying on gradient descent or backpropagation.
The libraries mimic PyTorch's sequential interface, allowing users to define models by stacking layers in order and performing uncertainty-aware Bayesian inference. Beyond epistemic uncertainty, it also allows quantifying heteroscedastic aleatoric uncertainty. cuTAGI's custom CPU/GPU kernels and distributed-data-parallel support via NCCL/MPI deliver competitive runtimes, while pyTAGI's pip-installable frontend and MIT-licensed GitHub repo facilitate community adoption and extension. Version 0.2.1 already supports a comprehensive suite of layers and activations; future work will add eager execution, further kernel optimizations, attention mechanisms, and advanced covariance factorization. Together, py/cuTAGI offer an efficient, open-source foundation for the analytic treatment of Bayesian deep learning.

Схожие новости

#Наименование новостиТональностьИнформативностьДата публикации
1 Generative Bayesian Inference with GANs 06.6217-08-2026
2 Robust training of implicit generative models for multivariate and heavy-tailed distributions with an invariant statistical loss 01017-08-2026
3 End-to-End Deep Learning for Predicting Metric Space-Valued Outputs 010.6617-08-2026
4 Adaptive Nonparametric Perturbations of Parametric Models with Generalized Bayes 04.6217-08-2026
5 High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networks 08.717-08-2026
6 Mixing times of data-augmentation Gibbs samplers for high-dimensional probit regression 08.7817-08-2026
7 Approximation-Free Differentiable Oblique Decision Trees 09.617-08-2026
8 UQLM: A Python Package for Uncertainty Quantification in Large Language Models 010.2917-08-2026
9ctboost 0.1.56028.1814-08-2026
10torchrl-nightly 2026.7.200520-07-2026

Классификация: Пресс-релизы. Схожих патентов: 0. Схожих новостей: 10. Тональность: 0. Информативность: 10. Источник: jmlr.org.