lhnguyen102 / cuTAGI
CUDA implementation of Tractable Approximate Gaussian Inference
☆31Updated this week
Related projects ⓘ
Alternatives and complementary repositories for cuTAGI
- ☆38Updated 3 months ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆76Updated 8 months ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Pytorch implementation of VAEs for heterogeneous likelihoods.☆42Updated 2 years ago
- Running Jax in PyTorch Lightning☆82Updated 2 weeks ago
- ☆191Updated 6 months ago
- Sequence Modeling with Structured State Spaces☆60Updated 2 years ago
- Quantification of Uncertainty with Adversarial Models☆27Updated last year
- ☆41Updated last month
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆115Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆26Updated last month
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆122Updated 2 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆48Updated 11 months ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆21Updated 4 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆17Updated 2 weeks ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- A framework for conducting machine learning experiments in python