hardmaru / mdn_jax_tutorial
Mixture Density Networks (Bishop, 1994) tutorial in JAX
☆58Updated 4 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
- Normalizing Flows in Jax☆107Updated 4 years ago
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- fork of rl-baseline-zoo☆21Updated 4 years ago
- Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-…☆35Updated 5 years ago
- ☆153Updated 4 years ago
- ☆80Updated 3 years ago
- Documentation:☆119Updated last year
- Autoregressive Energy Machines☆77Updated 2 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 7 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Statistical adaptive stochastic optimization methods☆33Updated 4 years ago
- presentations☆44Updated 6 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆188Updated 2 years ago
- Python implementation of Bayesian Program Learning tools (with PyTorch)☆71Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- ☆80Updated last year
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- ☆29Updated 5 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆184Updated last year
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 8 months ago
- ☆68Updated last year
- Public Release of Plan2vec Implementation in pyTorch☆56Updated 2 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- ☆59Updated 6 years ago
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆39Updated 6 months ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago