AllanYangZhou / universal_neural_functionalLinks
☆51Updated 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
Sorting:
- The Energy Transformer block, in JAX☆59Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆93Updated 4 months ago
- NF-Layers for constructing neural functionals.☆87Updated last year
- ☆115Updated last month
- A centralized place for deep thinking code and experiments☆85Updated last year
- 🧱 Modula software package☆210Updated last week
- Scalable and Stable Parallelization of Nonlinear RNNS☆17Updated 6 months ago
- ☆53Updated 10 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- A simple library for scaling up JAX programs☆140Updated 9 months ago
- Implementation of PSGD optimizer in JAX☆34Updated 7 months ago
- ☆67Updated last year
- ☆53Updated last year
- LoRA for arbitrary JAX models and functions☆140Updated last year
- ☆43Updated 2 months ago
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆37Updated 2 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- 📄Small Batch Size Training for Language Models☆36Updated last week
- seqax = sequence modeling + JAX☆165Updated 2 weeks ago
- ☆31Updated 8 months ago
- ☆30Updated 4 months ago
- ☆232Updated 5 months ago
- Sparse and discrete interpretability tool for neural networks☆63Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆123Updated 7 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆105Updated 10 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆95Updated last month
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- ☆38Updated 3 weeks ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 4 years ago