kyunghyuncho / jax-practiceLinks
☆13Updated 4 years ago
Alternatives and similar repositories for jax-practice
Users that are interested in jax-practice are comparing it to the libraries listed below
Sorting:
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- ☆18Updated 3 years ago
- ☆45Updated 5 years ago
- ☆26Updated 6 years ago
- Pip-installable differentiable stacks in PyTorch!☆65Updated 4 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- ☆153Updated 5 years ago
- Silly twitter torch implementations.☆46Updated 2 years ago
- ☆64Updated 5 years ago
- ☆33Updated 4 years ago
- Reweighted Expectation Maximization☆29Updated 6 years ago
- Open source implementation of SeaRNN (ICLR 2018, https://openreview.net/forum?id=HkUR_y-RZ)☆48Updated 6 years ago
- Reference implementation of algorithms for reinforcement learning and Markov decision processes.☆12Updated 4 years ago
- An extension to Sacred for automated hyperparameter optimization.☆59Updated 7 years ago
- ☆30Updated 3 years ago
- learning to search in pytorch☆110Updated 5 years ago
- ☆12Updated 7 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- Normalizing Flows in Jax☆108Updated 4 years ago
- ☆108Updated 2 years ago
- ☆68Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 7 years ago
- A library for evaluating representations.☆76Updated 3 years ago
- ☆80Updated 3 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Computing various norms/measures on over-parametrized neural networks☆49Updated 6 years ago
- Recurrent Back Propagation, Back Propagation Through Optimization, ICML 2018☆42Updated 6 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 3 years ago