KindXiaoming / physics_of_skill_learningLinks
We study toy models of skill learning.
☆31Updated 8 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:
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆84Updated 11 months ago
- Implementation of Infini-Transformer in Pytorch☆113Updated 9 months ago
- ☆85Updated last year
- ☆48Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆119Updated 11 months ago
- ☆81Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated last year
- Implementation of the Llama architecture with RLHF + Q-learning☆167Updated 8 months ago
- Minimum Description Length probing for neural network representations☆20Updated 8 months ago
- A repository for research on medium sized language models.☆78Updated last year
- ☆38Updated last year
- GoldFinch and other hybrid transformer components☆45Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆102Updated 9 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆85Updated last month
- ☆56Updated 10 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆129Updated last year
- Implementation of Mind Evolution, Evolving Deeper LLM Thinking, from Deepmind☆57Updated 4 months ago
- Using FlexAttention to compute attention with different masking patterns☆45Updated last year
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆31Updated 5 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆49Updated 5 months ago
- Code for the paper "On the Expressivity Role of LayerNorm in Transformers' Attention" (Findings of ACL'2023)☆57Updated last year
- ☆32Updated 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…☆52Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆29Updated 7 months ago
- Anchored Preference Optimization and Contrastive Revisions: Addressing Underspecification in Alignment☆60Updated last year
- ☆58Updated last year
- Implementation of the general framework for AMIE, from the paper "Towards Conversational Diagnostic AI", out of Google Deepmind☆68Updated last year
- User-friendly implementation of the Mixture-of-Sparse-Attention (MoSA). MoSA selects distinct tokens for each head with expert choice rou…☆26Updated 5 months ago