A toolkit for scaling law research ⚖
☆69Jan 27, 2025Updated last year
Alternatives and similar repositories for chinchilla
Users that are interested in chinchilla are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- An open-source implementation of Scaling Laws for Neural Language Models using nanoGPT☆58Dec 8, 2023Updated 2 years ago
- Mamba support for transformer lens☆20Sep 17, 2024Updated 2 years ago
- PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu☆80Dec 3, 2024Updated last year
- ☆65Apr 9, 2024Updated 2 years ago
- Utilities for PyTorch distributed☆26Feb 27, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- PyTorch and NNsight implementation of AtP* (Kramar et al 2024, DeepMind)☆21Jan 19, 2025Updated last year
- WIP☆96Aug 13, 2024Updated 2 years ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- This repository houses materials consulted by the instructors☆12Jan 8, 2022Updated 4 years ago
- [WIP] Transformer to embed Danbooru labelsets☆13Mar 31, 2024Updated 2 years ago
- A tiny easily hackable implementation of a feature dashboard.☆18Oct 21, 2025Updated 11 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆11Aug 29, 2023Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆54May 20, 2024Updated 2 years ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Apr 17, 2024Updated 2 years ago
- [NeurIPS'24 Spotlight] Observational Scaling Laws☆63Oct 2, 2024Updated 2 years ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- ☆18Mar 18, 2024Updated 2 years ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆78Apr 2, 2024Updated 2 years ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆135Dec 3, 2024Updated last year
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- ☆20Oct 11, 2023Updated 2 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- This is the official repository for the "Towards Vision-Language Mechanistic Interpretability: A Causal Tracing Tool for BLIP" paper acce…☆25Feb 16, 2026Updated 7 months ago
- A library for unit scaling in PyTorch☆136Jul 11, 2025Updated last year
- Here we collect trick questions and failed tasks for open source LLMs to improve them.☆32Apr 20, 2023Updated 3 years ago
- [ICLR'25] "Understanding Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing" by Peihao Wang, Ruisi Cai, Yue…☆18Mar 21, 2025Updated last year
- seqax = sequence modeling + JAX☆199Jul 23, 2025Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆604Aug 14, 2026Updated last month
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆77Aug 2, 2024Updated 2 years ago
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- A repo to do interpretability of pre-trained acoustic models☆15Oct 15, 2023Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆23Apr 22, 2024Updated 2 years ago
- JAX/Flax implementation of the Hyena Hierarchy☆35Apr 27, 2023Updated 3 years ago
- Evaluating the Mamba architecture on the Othello game☆50Apr 25, 2024Updated 2 years ago
- ☆63Jul 9, 2024Updated 2 years ago
- Triton-based implementation of Sparse Mixture of Experts.☆279Oct 3, 2025Updated last year
- ☆67Dec 8, 2023Updated 2 years ago
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆11Dec 30, 2024Updated last year