google-deepmind / neural_testbedLinks
☆192Updated 2 months ago
Alternatives and similar repositories for neural_testbed
Users that are interested in neural_testbed are comparing it to the libraries listed below
Sorting:
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆115Updated last week
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- ☆107Updated 3 years ago
- Hierarchical Associative Memory User Experience☆103Updated 2 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆73Updated 2 years ago
- ☆312Updated 6 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆206Updated last year
- A collection of meta-learning algorithms in Jax☆23Updated 3 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆57Updated 4 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- ☆31Updated 3 years ago
- ☆54Updated last year
- Jax SSM Library☆49Updated 2 years ago
- Reinforcement learning library in JAX.☆100Updated last year
- Bayesianize: A Bayesian neural network wrapper in pytorch☆89Updated last year
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- ☆158Updated last year
- ☆26Updated 3 years ago
- Open source code for EigenGame.☆30Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆46Updated 4 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 2 years ago