lucidrains / scaling-vin-pytorch
Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
☆36Updated 6 months ago
Alternatives and similar repositories for scaling-vin-pytorch:
Users that are interested in scaling-vin-pytorch are comparing it to the libraries listed below
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆55Updated 10 months ago
- ☆30Updated 4 months ago
- ☆31Updated 11 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 3 months ago
- Explorations into improving ViTArc with Slot Attention☆38Updated 5 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 2 months ago
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆17Updated 9 months ago
- Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise☆32Updated 6 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated last year
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year
- ☆21Updated 5 months ago
- ☆24Updated 5 months ago
- ☆33Updated 6 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆15Updated 8 months ago
- Official PyTorch Implementation for Task Vectors are Cross-Modal☆22Updated 3 months ago
- ☆27Updated 8 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆53Updated 7 months ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆13Updated 2 weeks ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 3 months ago
- FID computation in Jax/Flax.☆27Updated 8 months ago
- ☆30Updated 5 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 9 months ago
- ☆33Updated 2 months ago