apple / ml-dataset-decompositionLinks
Official repo of dataset-decomposition paper [NeurIPS 2024]
☆19Updated 8 months ago
Alternatives and similar repositories for ml-dataset-decomposition
Users that are interested in ml-dataset-decomposition are comparing it to the libraries listed below
Sorting:
- A repository for research on medium sized language models.☆77Updated last year
- Common tools for data processing☆20Updated 2 weeks ago
- GoldFinch and other hybrid transformer components☆45Updated last year
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆84Updated 10 months ago
- Language models scale reliably with over-training and on downstream tasks☆99Updated last year
- Long Context Extension and Generalization in LLMs☆60Updated last year
- Repository for the Q-Filters method (https://arxiv.org/pdf/2503.02812)☆35Updated 6 months ago
- Official code repo for paper "Great Memory, Shallow Reasoning: Limits of kNN-LMs"☆23Updated 4 months ago
- ☆85Updated last year
- Reference implementation for Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model☆44Updated last year
- ☆39Updated last year
- Simple and efficient pytorch-native transformer training and inference (batched)☆78Updated last year
- Pytorch implementation for "Compressed Context Memory For Online Language Model Interaction" (ICLR'24)☆62Updated last year
- [ICLR 2025] Monet: Mixture of Monosemantic Experts for Transformers☆72Updated 3 months ago
- Minimum Description Length probing for neural network representations☆18Updated 7 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Awesome Triton Resources☆33Updated 5 months ago
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆34Updated last month
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆48Updated 5 months ago
- ☆53Updated 10 months ago
- Using FlexAttention to compute attention with different masking patterns☆44Updated last year
- ☆46Updated last year
- JAX Scalify: end-to-end scaled arithmetics☆16Updated 10 months ago
- ☆86Updated 8 months ago
- Here we will test various linear attention designs.☆62Updated last year
- Official implementation of the ICML 2024 paper RoSA (Robust Adaptation)☆44Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆70Updated last year
- Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…☆53Updated last year
- Exploration of automated dataset selection approaches at large scales.☆47Updated 6 months ago
- DPO, but faster 🚀☆44Updated 9 months ago