rbalestr-lab / stable-sslLinks
☆18Updated 2 weeks ago
Alternatives and similar repositories for stable-ssl
Users that are interested in stable-ssl are comparing it to the libraries listed below
Sorting:
- Scalable and Stable Parallelization of Nonlinear RNNS☆16Updated 5 months ago
- ☆104Updated 2 weeks ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆30Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- ☆150Updated 10 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆91Updated 2 weeks ago
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- Code for the paper: Rotating Features for Object Discovery☆52Updated 10 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 11 months ago
- ☆53Updated 8 months ago
- ☆65Updated last year
- ☆51Updated last year
- Minimal but scalable implementation of large language models in JAX☆35Updated 7 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆87Updated 3 months ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆20Updated last year
- ☆39Updated 3 years ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆27Updated 3 weeks ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆66Updated 5 months ago
- ☆15Updated 2 years ago
- 🧱 Modula software package☆200Updated 3 months ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆37Updated last month
- ☆27Updated last year
- ☆13Updated last year
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆82Updated 3 weeks ago
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆140Updated last month
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆126Updated last month
- ☆270Updated 11 months ago
- Focused on fast experimentation and simplicity☆75Updated 6 months ago
- ☆31Updated 7 months ago
- Code for minimum-entropy coupling.☆32Updated last year