arogozhnikov / readable_capsnet
Blazingly fast capsule networks in 75 lines of pytorch+einops
☆26Updated 3 years ago
Alternatives and similar repositories for readable_capsnet:
Users that are interested in readable_capsnet are comparing it to the libraries listed below
- ☆30Updated 2 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- ☆45Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- The stand-alone training engine module for the ALOHA.eu project.☆15Updated 5 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆39Updated 5 years ago
- Documentation:☆119Updated last year
- Code for "Aggregated Momentum: Stability Through Passive Damping", Lucas et al. 2018☆34Updated 6 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago
- Yet Another Neural Network Library 🤔☆25Updated 9 months ago
- ☆153Updated 4 years ago
- torch-inspect -- collection of utility functions to inspect low level information of neural network for PyTorch☆17Updated last year
- Experiment orchestration☆103Updated 4 years ago
- notebooks of cool EBM visualizations☆15Updated 3 years ago
- ☆108Updated 2 years ago
- Repo for open sourcing the NAMs.☆25Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆52Updated 3 years ago
- A tiny Catalyst-like experiment runner framework on top of micrograd.☆51Updated 4 years ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 4 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- PyTorch optimizers with sparse momentum and weight decay☆9Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- ☆11Updated 4 years ago
- Kervolution implementation using TF2.0☆20Updated 2 years ago
- Implementation and demonstration of backdrop in pytorch. Code and demonstration of GP dataset generator.☆68Updated 6 years 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☆43Updated 3 years ago
- A supplementary code for Beyond Vector Spaces: Compact Data Representation as Differentiable Weighted Graphs.☆47Updated 5 years ago