KindXiaoming / physics_of_skill_learningLinks
We study toy models of skill learning.
☆28Updated 5 months ago
Alternatives and similar repositories for physics_of_skill_learning
Users that are interested in physics_of_skill_learning are comparing it to the libraries listed below
Sorting:
- ☆79Updated 10 months ago
- Here we will test various linear attention designs.☆59Updated last year
- This repo is based on https://github.com/jiaweizzhao/GaLore☆28Updated 9 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Minimum Description Length probing for neural network representations☆18Updated 4 months ago
- Unofficial Implementation of Selective Attention Transformer☆17Updated 7 months ago
- JAX Scalify: end-to-end scaled arithmetics☆16Updated 7 months ago
- ☆37Updated last year
- Using FlexAttention to compute attention with different masking patterns☆44Updated 9 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 3 months ago
- ☆32Updated 5 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆66Updated 9 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 5 months ago
- ☆31Updated last year
- Implementation of Infini-Transformer in Pytorch☆111Updated 5 months ago
- Official code for the paper "Attention as a Hypernetwork"☆39Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆119Updated 8 months ago
- Explorations into adversarial losses on top of autoregressive loss for language modeling☆37Updated 4 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆55Updated 10 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 5 months ago
- Exploration of automated dataset selection approaches at large scales.☆45Updated 3 months ago
- [ICLR 2025] Monet: Mixture of Monosemantic Experts for Transformers☆69Updated this week
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆47Updated 2 months ago
- A basic pure pytorch implementation of flash attention☆16Updated 7 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆74Updated 7 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- ☆81Updated 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…☆54Updated last year
- GoldFinch and other hybrid transformer components☆45Updated 11 months ago