DPBayes / jax-chacha-prng
A cryptographically-secure pseudo-random number generator for JAX based on the 20 round ChaCha cipher.
☆12Updated 6 months ago
Related projects: ⓘ
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 5 years ago
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- Graph Learning with JAX☆14Updated 2 years ago
- ☆12Updated 3 years ago
- ☆31Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆41Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- Tools for JAX☆44Updated last week
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- ☆78Updated 3 years ago
- ☆33Updated 3 years ago
- Tensor Parallelism with JAX + Shard Map☆10Updated 11 months ago
- Variational Walkback, NIPS'17☆28Updated 6 years ago
- ☆18Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Modelling epidemiological dynamics and performing inference in these models☆27Updated 3 years ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆13Updated 7 months ago
- Quasi-Newton Algorithm for Stochastic Optimization☆10Updated 2 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆13Updated 4 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 4 years ago
- "Variational inference tools to leverage estimator sensitivity."☆16Updated last year
- ☆17Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Updated 3 years ago