yashbonde / rasp
Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.
☆51Updated 3 years ago
Alternatives and similar repositories for rasp:
Users that are interested in rasp are comparing it to the libraries listed below
- Named Tensors for Legible Deep Learning in JAX☆167Updated this week
- ☆19Updated last year
- Language-annotated Abstraction and Reasoning Corpus☆83Updated last year
- A domain-specific probabilistic programming language for modeling and inference with language models☆118Updated last year
- Learning Universal Predictors☆76Updated 7 months ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- seqax = sequence modeling + JAX☆150Updated last week
- Code associated to papers on superposition (in ML interpretability)☆28Updated 2 years ago
- ☆49Updated last year
- ☆20Updated 11 months ago
- Materials for ConceptARC paper☆89Updated 4 months ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆61Updated 3 years ago
- Experiments for efforts to train a new and improved t5☆77Updated 11 months ago
- Inference code for LLaMA models in JAX☆116Updated 10 months ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated 9 months ago
- Resources from the EleutherAI Math Reading Group☆53Updated 3 weeks ago
- Silly twitter torch implementations.☆46Updated 2 years ago
- Neural theorem proving tutorial, version II☆34Updated 10 months ago
- Notebooks accompanying Anthropic's "Toy Models of Superposition" paper☆117Updated 2 years ago
- Harmonic Datasets☆36Updated 8 months ago
- Code for minimum-entropy coupling.☆31Updated 8 months ago
- ☆62Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 2 years ago
- ☆55Updated 4 months ago
- Automatically take good care of your preemptible TPUs☆36Updated last year
- An environment for learning formal mathematical reasoning from scratch☆65Updated 7 months ago
- Mechanistic Interpretability for Transformer Models☆50Updated 2 years ago
- If it quacks like a tensor...☆57Updated 4 months ago
- gzip Predicts Data-dependent Scaling Laws☆34Updated 9 months ago