lhnguyen102 / cuTAGI
CUDA implementation of Tractable Approximate Gaussian Inference
☆31Updated last week
Alternatives and similar repositories for cuTAGI:
Users that are interested in cuTAGI are comparing it to the libraries listed below
- Running Jax in PyTorch Lightning☆83Updated last month
- Parameter-Free Optimizers for Pytorch☆109Updated 8 months ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆128Updated 2 years ago
- Image augmentation library for Jax☆37Updated 9 months ago
- ☆191Updated 8 months ago
- Hierarchical Associative Memory User Experience☆94Updated last year
- Materials of the Nordic Probabilistic AI School 2023.☆85Updated last year
- TorchDR - PyTorch Dimensionality Reduction☆73Updated this week
- Lightweight Cluster/Cloud VM Job Management 🚀☆39Updated 4 months ago
- ☆42Updated 5 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆148Updated last month
- A library for uncertainty quantification based on PyTorch☆120Updated 3 years ago
- Training and evaluating NBM and SPAM for interpretable machine learning.☆75Updated last year
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 8 months ago
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- ☆111Updated last week
- ☆41Updated 2 months ago
- Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)☆12Updated last year
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Quantification of Uncertainty with Adversarial Models☆27Updated last year
- Lightweight ML Experiment Logging 📖☆78Updated 4 months ago
- Pytorch implementation of VAEs for heterogeneous likelihoods.☆42Updated 2 years ago
- Lightning-like training API for JAX with Flax☆36Updated last month
- My implementation of DeepMind's Perceiver☆61Updated 3 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆41Updated last year
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated 10 months ago
- Neural Networks for JAX☆83Updated 3 months ago