EleutherAI / equivariance
A framework for implementing equivariant DL
☆10Updated 3 years ago
Alternatives and similar repositories for equivariance:
Users that are interested in equivariance are comparing it to the libraries listed below
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆23Updated 4 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆14Updated 3 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 3 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- ☆18Updated 2 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- AdaCat☆49Updated 2 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Updated 6 years ago
- Public Release of Plan2vec Implementation in pyTorch☆56Updated 2 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- ☆12Updated 2 years ago
- ☆24Updated 5 years ago
- ☆33Updated 4 years ago
- Using Rainbow implementation in Chainer RL for Slime Volleyball Pixel Environment☆23Updated 4 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆49Updated 3 years ago
- A simple Transformer where the softmax has been replaced with normalization☆19Updated 4 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆47Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- ☆15Updated 2 years ago
- ☆22Updated 3 years ago
- ☆21Updated last year
- ☆26Updated 2 years ago
- ☆30Updated 2 years ago
- ☆24Updated 8 months ago
- Variational Reinforcement Learning☆16Updated 6 months ago