google-deepmind / neural_testbedLinks
☆191Updated 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:
- Hierarchical Associative Memory User Experience☆103Updated last month
- ☆115Updated last week
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆158Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆206Updated 11 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆108Updated last year
- ☆106Updated 3 years ago
- Neural Turing Machines in pytorch☆48Updated 3 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆73Updated 2 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- ☆54Updated last year
- Image augmentation library for Jax☆39Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆154Updated 10 months ago
- Easy Hypernetworks in Pytorch and Jax☆104Updated 2 years ago
- ☆312Updated 5 months ago
- Reinforcement learning library in JAX.☆100Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated last year
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Updated 3 years ago
- A functional training loops library for JAX☆88Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Tutorial on amortized optimization for learning to optimize over continuous domains☆243Updated 5 months ago
- Deep Probabilistic Programming Course @ DIKU☆58Updated 4 years ago