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☆56Dec 8, 2023Updated 2 years ago
- Mamba support for transformer lens☆20Sep 17, 2024Updated last year
- PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu☆79Dec 3, 2024Updated last year
- ☆65Apr 9, 2024Updated 2 years ago
- Utilities for PyTorch distributed☆26Feb 27, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- PyTorch and NNsight implementation of AtP* (Kramar et al 2024, DeepMind)☆20Jan 19, 2025Updated last year
- WIP☆96Aug 13, 2024Updated 2 years ago
- A blog where I write about research papers and blog posts I read.☆12Nov 20, 2024Updated last year
- 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☆13Jan 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.☆17Oct 21, 2025Updated 9 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆11Aug 29, 2023Updated 2 years ago
- Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …☆15Mar 11, 2024Updated 2 years ago
- ☆54May 20, 2024Updated 2 years ago
- Minimum Description Length probing for neural network representations☆20Jan 28, 2025Updated last year
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Apr 17, 2024Updated 2 years ago
- [NeurIPS'24 Spotlight] Observational Scaling Laws☆61Oct 2, 2024Updated last year
- 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
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- 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
- This is the official repository for the "Towards Vision-Language Mechanistic Interpretability: A Causal Tracing Tool for BLIP" paper acce…☆25Feb 16, 2026Updated 6 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☆198Jul 23, 2025Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆606Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Aug 12, 2023Updated 3 years ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆75Aug 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
- ☆22Apr 22, 2024Updated 2 years ago
- JAX/Flax implementation of the Hyena Hierarchy☆35Apr 27, 2023Updated 3 years ago
- ☆61Jul 9, 2024Updated 2 years ago