lhnguyen102 / cuTAGI
CUDA implementation of Tractable Approximate Gaussian Inference
☆35Updated this week
Alternatives and similar repositories for cuTAGI:
Users that are interested in cuTAGI are comparing it to the libraries listed below
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 8 months ago
- A library for uncertainty quantification based on PyTorch☆123Updated 3 years ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆116Updated 3 years ago
- Parameter-Free Optimizers for Pytorch☆122Updated 11 months ago
- Normalizing Flows using JAX☆83Updated last year
- Normalizing Flows with a resampled base distribution☆46Updated 2 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆117Updated last year
- Image augmentation library for Jax☆39Updated 11 months ago
- ☆45Updated last week
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated this week
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Implementation of normalizing flows from 1d to Nd☆32Updated 4 years ago
- Running Jax in PyTorch Lightning☆92Updated 3 months ago
- ☆192Updated 10 months ago
- Simple (and cheap!) neural network uncertainty estimation☆63Updated this week
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 10 months ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 2 years ago
- ☆113Updated this week
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆102Updated last year
- ☆240Updated 2 years ago
- Neat Bayesian machine learning examples☆55Updated 2 months ago
- Documentation:☆119Updated last year
- Hierarchical Associative Memory User Experience☆100Updated last year
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- A functional training loops library for JAX☆86Updated last year
- Pytorch-like dataloaders for JAX.☆77Updated 5 months ago
- Riemannian Optimization Using JAX☆48Updated last year