hardmaru / mdn_jax_tutorial
Mixture Density Networks (Bishop, 1994) tutorial in JAX
☆57Updated 4 years ago
Related projects: ⓘ
- Normalizing Flows in Jax☆105Updated 4 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- fork of rl-baseline-zoo☆21Updated 4 years ago
- ☆78Updated 3 years ago
- 2019 talk at GECCO☆68Updated 5 years ago
- ☆156Updated 4 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 6 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆50Updated 3 years ago
- Example python package with pybind11 cpp extension☆57Updated 3 years ago
- Autoregressive Energy Machines☆77Updated last year
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-…☆34Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- presentations☆44Updated 5 years ago
- ☆45Updated 4 years ago
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- ☆64Updated 10 months ago
- ☆29Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆120Updated 2 months ago
- Graduate topics course on learning discrete latent structure.☆66Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- Train self-modifying neural networks with neuromodulated plasticity☆77Updated 4 years ago
- ☆80Updated 11 months ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- ☆82Updated 4 years ago
- Library for learning and inference with Sum-product Networks utilizing TensorFlow 2.x and Keras☆47Updated 3 years ago
- Neural Turing Machines in pytorch☆46Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆41Updated 3 years ago
- ☆10Updated this week