konstantinosKokos / apeLinks
🧮 Algebraic Positional Encodings.
☆16Updated last week
Alternatives and similar repositories for ape
Users that are interested in ape are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆59Updated last year
- ☆32Updated 10 months ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆34Updated 11 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- ☆30Updated 5 months ago
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 3 months ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆59Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆20Updated this week
- A Scalable Approximate Method for Probabilistic Neurosymbolic Inference☆16Updated 7 months ago
- Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021☆13Updated 3 years ago
- RWKV model implementation☆38Updated 2 years ago
- ☆56Updated 10 months ago
- Blog post☆17Updated last year
- ☆22Updated 3 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆12Updated 4 years ago
- ☆36Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆33Updated 9 months ago
- ☆13Updated 3 months ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆17Updated 3 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆20Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated 2 years ago
- ☆34Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19Updated 3 months ago
- ☆11Updated last year
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆84Updated 2 years ago
- An annotated implementation of the Hyena Hierarchy paper☆33Updated 2 years ago