ejmichaud / grokking-squared
☆22Updated last year
Related projects ⓘ
Alternatives and complementary repositories for grokking-squared
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆57Updated last year
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆80Updated 2 years ago
- Code for minimum-entropy coupling.☆30Updated 4 months ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆76Updated last year
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆44Updated 2 years ago
- The official repository for our paper "Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Masks". We…☆46Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆52Updated last year
- ☆21Updated 2 years ago
- Bayesian model reduction for probabilistic machine learning☆10Updated 9 months ago
- ☆49Updated 4 years ago
- Deep Learning & Information Bottleneck☆50Updated last year
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆17Updated 2 years ago
- Code for the paper: "Independent mechanism analysis, a new concept?"☆25Updated last year
- Omnigrok: Grokking Beyond Algorithmic Data☆50Updated last year
- ☆25Updated last month
- Laplace Redux -- Effortless Bayesian Deep Learning☆38Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- The Energy Transformer block, in JAX☆53Updated 11 months ago
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆63Updated 2 years ago
- ☆26Updated last year
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆34Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆79Updated 9 months ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆40Updated last year
- ☆59Updated 3 years ago
- Structured Neural Networks☆13Updated 6 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆38Updated 9 months ago