hardmaru / mdn_jax_tutorialLinks
Mixture Density Networks (Bishop, 1994) tutorial in JAX
☆61Updated 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
Sorting:
- ☆153Updated 5 years ago
- 2019 talk at GECCO☆68Updated 6 years ago
- Normalizing Flows in Jax☆107Updated 5 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Updated 3 years ago
- Autoregressive Energy Machines☆78Updated 3 years ago
- fork of rl-baseline-zoo☆21Updated 5 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆192Updated 2 years ago
- ☆182Updated last year
- Neural Turing Machines in pytorch☆48Updated 3 years ago
- Pip-installable differentiable stacks in PyTorch!☆65Updated 5 years ago
- Python implementation of GLN in different frameworks☆97Updated 5 years ago
- ☆80Updated 2 years ago
- ☆71Updated 2 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- ☆172Updated last year
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆53Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 4 months ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- ☆45Updated 6 years ago
- ☆83Updated 5 years ago
- The Differentiable Cross-Entropy Method☆124Updated 5 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆14Updated 6 years ago
- Train self-modifying neural networks with neuromodulated plasticity☆78Updated 6 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆109Updated last year
- Adapting the AlphaZero algorithm to remove the need of execution traces to train NPI.☆79Updated 2 years ago
- ☆42Updated 5 years ago