AllanYangZhou / universal_neural_functional
☆49Updated last year
Alternatives and similar repositories for universal_neural_functional:
Users that are interested in universal_neural_functional are comparing it to the libraries listed below
- ☆52Updated 5 months ago
- Deep Networks Grok All the Time and Here is Why☆31Updated 10 months ago
- Code associated to papers on superposition (in ML interpretability)☆28Updated 2 years ago
- The Energy Transformer block, in JAX☆56Updated last year
- A centralized place for deep thinking code and experiments☆82Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆72Updated last week
- 🧱 Modula software package☆173Updated 2 weeks ago
- ☆58Updated 9 months ago
- ☆25Updated 3 weeks ago
- Implementation of PSGD optimizer in JAX☆29Updated 2 months ago
- NF-Layers for constructing neural functionals.☆82Updated last year
- ☆53Updated last year
- ☆25Updated last year
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆35Updated 2 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 4 months ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- ☆26Updated 8 months ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 6 months ago
- ☆31Updated 11 months ago
- Code for minimum-entropy coupling.☆31Updated 8 months ago
- Code release for REPAIR: REnormalizing Permuted Activations for Interpolation Repair☆47Updated last year
- Universal Neurons in GPT2 Language Models☆27Updated 9 months ago
- Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"☆49Updated 9 months ago
- ☆30Updated 4 months ago
- ☆30Updated 5 months ago
- ☆41Updated 4 months ago