nvecoven / BRC
A repository containing the code for the Bistable Recurrent Cell
☆47Updated 4 years ago
Alternatives and similar repositories for BRC:
Users that are interested in BRC are comparing it to the libraries listed below
- Pytorch implementation of bistable recurrent cell with baseline comparisons.☆25Updated last year
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 8 months ago
- Python implementation of GLN in different frameworks☆98Updated 4 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Deep Probabilistic Programming Course @ DIKU☆57Updated 4 years ago
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆41Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'☆31Updated 10 months ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- ☆42Updated 4 years ago
- fork of rl-baseline-zoo☆21Updated 4 years ago
- My implementation of DeepMind's Perceiver☆63Updated 3 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆103Updated 3 years ago
- ☆153Updated 4 years ago
- Neural Turing Machines in pytorch☆48Updated 3 years ago
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆39Updated 6 months ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Code for scaling Transformers☆26Updated 4 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 3 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- ☆49Updated 4 years ago
- ☆31Updated 3 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 8 months ago