google-deepmind / neural_testbedLinks
☆194Updated last week
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.☆39Updated 3 years ago
- ☆118Updated last month
- Image augmentation library for Jax☆41Updated last year
- Hierarchical Associative Memory User Experience☆106Updated this week
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆61Updated 3 years ago
- Some small scale experiments for my blog posts 📝☆80Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆47Updated 4 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Updated 5 years ago
- ☆112Updated 3 years ago
- Reinforcement learning library in JAX.☆100Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Updated 5 years ago
- Neural Turing Machines in pytorch☆48Updated 4 years ago
- ☆312Updated last week
- Easy Hypernetworks in Pytorch and Jax☆106Updated 2 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆208Updated last year
- Neural Networks for JAX☆84Updated last year
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 3 years ago
- Composable kernels for scikit-learn implemented in JAX.☆45Updated 5 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Fully differentiable RL environments, written in Ivy.☆66Updated 2 years ago
- ☆162Updated 2 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 3 years ago
- ☆26Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- ☆100Updated 4 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- Jax SSM Library☆48Updated 3 years ago
- A functional training loops library for JAX☆88Updated last year
- Parameter-Free Optimizers for Pytorch☆130Updated last year