rbalestr-lab / stable-sslLinks
☆17Updated this week
Alternatives and similar repositories for stable-ssl
Users that are interested in stable-ssl are comparing it to the libraries listed below
Sorting:
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆86Updated 2 months ago
- Minimal GPT (~350 lines with a simple task to test it)☆62Updated 5 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆29Updated last year
- ☆27Updated last year
- Implementation of the proposed Spline-Based Transformer from Disney Research☆92Updated 6 months ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆26Updated this week
- Scalable and Stable Parallelization of Nonlinear RNNS☆15Updated 4 months ago
- ☆33Updated 5 months ago
- Deep Networks Grok All the Time and Here is Why☆36Updated last year
- ☆150Updated 9 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 4 months ago
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆139Updated 2 weeks ago
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models☆58Updated 3 months ago
- ☆51Updated last year
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 10 months ago
- A basic pure pytorch implementation of flash attention☆16Updated 7 months ago
- ☆32Updated last year
- ☆15Updated 2 years ago
- Code for the paper "Function-Space Learning Rates"☆20Updated this week
- ☆53Updated 8 months ago
- ☆65Updated 11 months ago
- ☆26Updated this week
- ☆53Updated last year
- Focused on fast experimentation and simplicity☆73Updated 5 months ago
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated last week
- ☆23Updated 2 weeks ago
- ViT Prisma is a mechanistic interpretability library for Vision and Video Transformers (ViTs).☆250Updated this week
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆63Updated this week
- ☆27Updated 10 months ago
- Brain-like variational inference☆51Updated 2 weeks ago