lhnguyen102 / cuTAGILinks
CUDA implementation of Tractable Approximate Gaussian Inference
☆41Updated last week
Alternatives and similar repositories for cuTAGI
Users that are interested in cuTAGI are comparing it to the libraries listed below
Sorting:
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Hierarchical Associative Memory User Experience☆104Updated 2 months ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆118Updated 4 years ago
- ☆192Updated 3 months ago
- Materials of the Nordic Probabilistic AI School 2022.☆179Updated 3 years ago
- Bayesian and Maximum Likelihood Implementation of the Normalizing Flow Network (NFN): https://arxiv.org/abs/1907.08982☆21Updated 4 years ago
- Pre-trained Gaussian processes for Bayesian optimization☆97Updated 5 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆81Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆142Updated 3 months ago
- Running Jax in PyTorch Lightning☆113Updated 9 months ago
- A library for uncertainty quantification based on PyTorch☆122Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆130Updated 3 years ago
- Official code for UnICORNN (ICML 2021)☆28Updated 4 years ago
- ☆115Updated 3 weeks ago
- Materials of the Nordic Probabilistic AI School 2023.☆90Updated last year
- ☆311Updated 7 months ago
- ☆154Updated 3 years ago
- Image augmentation library for Jax☆40Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆309Updated 3 months ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 5 months ago
- Bayesian algorithm execution (BAX)☆50Updated 4 years ago
- Implementation of normalizing flows from 1d to Nd☆36Updated 4 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆74Updated 2 years ago
- Neural Networks for JAX☆84Updated last year
- Training and evaluating NBM and SPAM for interpretable machine learning.☆78Updated 2 years ago
- Bayesian inference with Python and Jax.☆34Updated 2 years ago
- Simple (and cheap!) neural network uncertainty estimation☆70Updated 4 months ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago