ofnote / tsanley
A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
☆40Updated 5 years ago
Alternatives and similar repositories for tsanley:
Users that are interested in tsanley are comparing it to the libraries listed below
- ☆30Updated 3 years ago
- ☆45Updated 5 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- ☆108Updated 2 years ago
- A supplementary code for Beyond Vector Spaces: Compact Data Representation as Differentiable Weighted Graphs.☆47Updated 5 years ago
- The stand-alone training engine module for the ALOHA.eu project.☆15Updated 5 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Code for "Aggregated Momentum: Stability Through Passive Damping", Lucas et al. 2018☆34Updated 6 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- ☆42Updated 4 years ago
- Variational Walkback, NIPS'17☆28Updated 7 years ago
- ☆10Updated 5 years ago
- Generative Latent Attentive Sampler☆26Updated 7 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- stochs: fast stochastic solvers for machine learning in C++ and Cython☆26Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Statistical adaptive stochastic optimization methods☆32Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Distributed Bayesian Optimization☆23Updated 4 years ago
- Cellular automaton-based calculus for the masses☆24Updated 6 years ago
- Reference implementation of algorithms for reinforcement learning and Markov decision processes.☆12Updated 4 years ago
- Pip-installable differentiable stacks in PyTorch!☆65Updated 4 years ago
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- Research code need not be ugly.☆77Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Code to reproduce experiments appearing in the academic paper Lost Relatives of the Gumbel Trick☆17Updated 7 years ago
- ☆17Updated 7 years ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- Flexible Reinforcement Learning Framework with PyTorch☆22Updated 4 years ago