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 | 0 | 6.62 | 17-08-2026 |
| 2 | Robust training of implicit generative models for multivariate and heavy-tailed distributions with an invariant statistical loss | 0 | 10 | 17-08-2026 |
| 3 | End-to-End Deep Learning for Predicting Metric Space-Valued Outputs | 0 | 10.66 | 17-08-2026 |
| 4 | Adaptive Nonparametric Perturbations of Parametric Models with Generalized Bayes | 0 | 4.62 | 17-08-2026 |
| 5 | High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networks | 0 | 8.7 | 17-08-2026 |
| 6 | Mixing times of data-augmentation Gibbs samplers for high-dimensional probit regression | 0 | 8.78 | 17-08-2026 |
| 7 | Approximation-Free Differentiable Oblique Decision Trees | 0 | 9.6 | 17-08-2026 |
| 8 | UQLM: A Python Package for Uncertainty Quantification in Large Language Models | 0 | 10.29 | 17-08-2026 |
| 9 | ctboost 0.1.56 | 0 | 28.18 | 14-08-2026 |
| 10 | torchrl-nightly 2026.7.20 | 0 | 5 | 20-07-2026 |