galilai-group / llm-jepaLinks
☆152Updated 3 months ago
Alternatives and similar repositories for llm-jepa
Users that are interested in llm-jepa are comparing it to the libraries listed below
Sorting:
- Official JAX implementation of End-to-End Test-Time Training for Long Context☆102Updated last week
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆82Updated 7 months ago
- Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.☆133Updated 3 months ago
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆31Updated 8 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆118Updated 2 months ago
- Official repo of paper LM2☆46Updated 10 months ago
- Official implementation of Regularized Policy Gradient (RPG) (https://arxiv.org/abs/2505.17508)☆61Updated last week
- RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best…☆59Updated 9 months ago
- Geometric-Mean Policy Optimization☆94Updated last month
- RLP: Reinforcement as a Pretraining Objective☆220Updated 3 months ago
- ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"☆121Updated 6 months ago
- The official github repo for "Diffusion Language Models are Super Data Learners".☆215Updated 2 months ago
- Official Repository of Native Parallel Reasoner☆89Updated 2 weeks ago
- Landing repository for the paper "Predicting the Order of Upcoming Tokens Improves Language Modeling"☆41Updated 3 months ago
- [NeurIPS 2025 Oral] Exploring Diffusion Transformer Designs via Grafting☆69Updated 6 months ago
- NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…☆43Updated last year
- Flash Attention Triton kernel with support for second-order derivatives☆126Updated 2 weeks ago
- ☆19Updated 10 months ago
- Esoteric Language Models☆108Updated last month
- Defeating the Training-Inference Mismatch via FP16☆170Updated last month
- Large multi-modal models (L3M) pre-training.☆223Updated 3 months ago
- ☆72Updated 5 months ago
- Implementation of Mind Evolution, Evolving Deeper LLM Thinking, from Deepmind☆57Updated 7 months ago
- ☆29Updated last month
- 📄Small Batch Size Training for Language Models☆77Updated 3 months ago
- X-Reasoner: Towards Generalizable Reasoning Across Modalities and Domains☆50Updated 7 months ago
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆149Updated 3 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆134Updated 2 weeks ago
- Official implementation of paper "Think-at-Hard: Selective Latent Iterations to Improve Reasoning Language Models"☆56Updated 2 weeks ago
- This repo contains the source code for the paper "Evolution Strategies at Scale: LLM Fine-Tuning Beyond Reinforcement Learning"☆279Updated last month