hardmaru / mdn_jax_tutorial
Mixture Density Networks (Bishop, 1994) tutorial in JAX
☆59Updated 5 years ago
Alternatives and similar repositories for mdn_jax_tutorial:
Users that are interested in mdn_jax_tutorial are comparing it to the libraries listed below
- 2019 talk at GECCO☆68Updated 5 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- ☆153Updated 4 years ago
- fork of rl-baseline-zoo☆21Updated 4 years ago
- ☆80Updated 3 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 7 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 9 months ago
- presentations☆44Updated 6 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- ☆67Updated last year
- Neural Turing Machines in pytorch☆48Updated 3 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆189Updated 2 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-…☆35Updated 5 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- ☆45Updated 5 years ago
- Documentation:☆119Updated last year
- Interactively retrieve data from sacred experiments.☆82Updated last month
- Statistical adaptive stochastic optimization methods☆32Updated 5 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆47Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆14Updated 6 years ago
- Train self-modifying neural networks with neuromodulated plasticity☆76Updated 5 years ago
- ☆20Updated 5 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- An extension to Sacred for automated hyperparameter optimization.☆59Updated 7 years ago
- ☆26Updated 6 years ago