IBM / pylspack
A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
☆18Updated this week
Related projects: ⓘ
- Riemannian Optimization Using JAX☆44Updated 10 months ago
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆17Updated 2 months ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- A differentiation API for PyTorch☆29Updated 4 years ago
- Euclidean Wasserstein-2 optimal transportation☆43Updated last year
- Riemannian Convex Potential Maps☆68Updated last year
- ☆78Updated 3 years ago
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- Neural likelihood-free methods in PyTorch.☆38Updated 4 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- ☆50Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆94Updated 10 months ago
- This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.☆15Updated last year
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆15Updated 2 years ago
- ☆49Updated 3 years ago
- ☆31Updated 3 years ago
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆18Updated 5 years ago
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Updated 4 years ago
- ☆12Updated 3 years ago
- Hessian trace estimation using PyTorch and Hutch++☆18Updated 3 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 4 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆35Updated last year
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 3 years ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆16Updated 2 years ago
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22Updated 5 years ago
- Random feature latent variable models in Python☆21Updated last year
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago