stable-pretraining-v1: Foundation Model Research Made Simple
Not on the map. Why?
Abstract (arXiv)
Foundation models and self-supervised learning (SSL) have become central to modern AI, yet research in this area remains hindered by complex codebases, redundant re-implementations, and the heavy engineering burden of scaling experiments. We present stable-pretraining, a modular, extensible, and performance-optimized library built on top of PyTorch, Lightning, Hugging Face, and TorchMetrics. Unlike prior toolkits focused narrowly on reproducing state-of-the-art results, stable-pretraining is designed for flexibility and iteration speed: it unifies essential SSL utilities--including probes, collapse detection metrics, augmentation pipelines, and extensible evaluation routines--within a coherent and reliable framework. A central design principle is logging everything, enabling fine-grained visibility into training dynamics that makes debugging, monitoring, and reproducibility seamless. We validate the library by demonstrating its ability to generate new research insights with minimal overhead, including depthwise representation probing and the analysis of CLIP degradation under synthetic data finetuning. By lowering barriers to entry while remaining scalable to large experiments, stable-pretraining aims to accelerate discovery and expand the possibilities of foundation model research.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 13 of its 16 references that are in the map. First posted on arXiv 23 Nov 2025.
Builds on
13 of 16 references.
References · 13
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2025DINOv3cited by 1,297
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
- 2021Datasets: A Community Library for Natural Language Processingcited by 526
- 2022DiffusionDB: A Large-scale Prompt Gallery Dataset for Text-to-Image Generative Modelscited by 376
- 2023A Cookbook of Self-Supervised Learningcited by 248
- 2022RankMe: Assessing the downstream performance of pretrained self-supervised representations by their rankcited by 145
- 2021Solo-learn: A Library of Self-supervised Methods for Visual Representation Learningcited by 116
1 more
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixelscited by 237
- 2026Causal-JEPA: Learning World Models through Object-Level Latent Maskingcited by 25
- 2026A Lightweight Library for Energy-Based Joint-Embedding Predictive Architecturescited by 13
- 2026stable-worldmodel-v1: Reproducible World Modeling Research and Evaluationcited by 8
- 2026FF-JEPA: Long-Horizon Planning in World Models with Latent Plannerscited by 6
- 2026LeVLJEPA: End-to-End Vision-Language Pretraining Without Negativescited by 2
- 2026Generate in Reconstruction Space, Match in Semantic Space: Transport Geometry for One-Step Generationcited by 0
- 2026Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Moleculescited by 0
- 2026CF-JEPA: Improving Robustness of JEPA World Models via Controllability Factorizationcited by 0
- 2026CF-JEPA: Improving Robustness of JEPA World Models via Controllability Factorization30 Sep 2026
- 2026Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Molecules23 Aug 2026
- 2026LeVLJEPA: End-to-End Vision-Language Pretraining Without Negatives1 Jul 2026
- 2026FF-JEPA: Long-Horizon Planning in World Models with Latent Planners8 Jun 2026
- 2026Generate in Reconstruction Space, Match in Semantic Space: Transport Geometry for One-Step Generation30 May 2026
- 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels13 Mar 2026
- 2026Causal-JEPA: Learning World Models through Object-Level Latent Masking11 Feb 2026
- 2026stable-worldmodel-v1: Reproducible World Modeling Research and Evaluation9 Feb 2026
- 2026A Lightweight Library for Energy-Based Joint-Embedding Predictive Architectures3 Feb 2026