namedtensor / notation
☆108Updated 2 years ago
Alternatives and similar repositories for notation:
Users that are interested in notation are comparing it to the libraries listed below
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- ☆153Updated 4 years ago
- Silly twitter torch implementations.☆46Updated 2 years ago
- ☆30Updated 2 years ago
- ☆102Updated 4 years ago
- ☆45Updated 5 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- ☆64Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆52Updated 3 years ago
- ☆74Updated 5 years ago
- ☆33Updated 4 years ago
- Docs☆143Updated last month
- Check if you have training samples in your test set☆64Updated 2 years ago
- Repo for open sourcing the NAMs.☆25Updated 4 years ago
- ☆42Updated 3 years ago
- Pip-installable differentiable stacks in PyTorch!☆65Updated 4 years ago
- ☆38Updated 2 years ago
- Weekly humanitarian AI reading group at Mila.☆21Updated 5 years ago
- Functional machine learning for fun☆84Updated 3 years ago
- ☆83Updated 4 years ago
- Code for scaling Transformers☆26Updated 4 years ago
- ☆98Updated 3 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆39Updated 5 years ago
- A list of resources dedicated to compositionality☆14Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- ☆100Updated 6 months ago
- ☆11Updated 4 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆66Updated 2 years ago