ssokota / mec
Code for minimum-entropy coupling.
☆31Updated 7 months ago
Alternatives and similar repositories for mec:
Users that are interested in mec are comparing it to the libraries listed below
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆58Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- gzip Predicts Data-dependent Scaling Laws☆34Updated 8 months ago
- Scaling scaling laws with board games.☆47Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆33Updated last year
- Sparse and discrete interpretability tool for neural networks☆58Updated last year
- Clean RL implementation using MLX☆28Updated 11 months ago
- ☆13Updated 6 months ago
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆20Updated 5 months ago
- Learn online intrinsic rewards from LLM feedback☆34Updated last month
- Generative cellular automaton-like learning environments for RL.☆19Updated 2 weeks ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 8 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 3 weeks ago
- Automatically take good care of your preemptible TPUs☆36Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 months ago
- Repo to reproduce the First-Explore paper results☆37Updated last month
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- ☆15Updated 5 months ago
- ☆59Updated 2 years ago
- Minimum Description Length probing for neural network representations☆18Updated 2 weeks ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated last year
- The Energy Transformer block, in JAX☆55Updated last year
- ☆31Updated 2 years ago
- Portfolio REgret for Confidence SEquences☆14Updated 2 months ago
- ☆29Updated 4 months ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆20Updated 7 months ago
- ☆37Updated 2 years ago
- ☆40Updated 2 months ago
- PyTorch Package For Quasimetric Learning☆41Updated 3 months ago