lhnguyen102 / cuTAGILinks
CUDA implementation of Tractable Approximate Gaussian Inference
☆38Updated last week
Alternatives and similar repositories for cuTAGI
Users that are interested in cuTAGI are comparing it to the libraries listed below
Sorting:
- Running Jax in PyTorch Lightning☆102Updated 5 months ago
- ☆192Updated last month
- ☆51Updated last month
- A library for uncertainty quantification based on PyTorch☆121Updated 3 years ago
- Hierarchical Associative Memory User Experience☆100Updated last year
- Neat Bayesian machine learning examples☆58Updated 4 months ago
- Implementation of normalizing flows from 1d to Nd☆34Updated 4 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆78Updated 11 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated last week
- Simple (and cheap!) neural network uncertainty estimation☆66Updated last week
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆129Updated last year
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆118Updated 4 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆106Updated 10 months ago
- Neural Networks for JAX☆84Updated 8 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Pytorch-like dataloaders for JAX.☆85Updated last week
- Riemannian Optimization Using JAX☆49Updated last year
- Materials of the Nordic Probabilistic AI School 2023.☆90Updated last year
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆27Updated last month
- Lightning-like training API for JAX with Flax☆38Updated 5 months ago
- Implementation of Action Matching☆44Updated 2 years ago
- ☆48Updated 4 months ago
- Normalizing Flows using JAX☆83Updated last year
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Updated last year
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆21Updated 4 years ago