google-deepmind / neural_testbedLinks
☆192Updated 5 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:
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- ☆118Updated last month
- Meta-learning inductive biases in the form of useful conserved quantities.☆38Updated 3 years ago
- Composable kernels for scikit-learn implemented in JAX.☆45Updated 5 years ago
- ☆26Updated 3 years ago
- Image augmentation library for Jax☆40Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 5 years ago
- Hierarchical Associative Memory User Experience☆105Updated 3 weeks ago
- ☆312Updated 9 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- Open source code for EigenGame.☆33Updated 2 years ago
- ☆161Updated last year
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- Neural Turing Machines in pytorch☆48Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆117Updated 3 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆20Updated 5 years ago
- Normalizing Flows using JAX☆85Updated 2 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆77Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆61Updated 5 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆47Updated 4 years ago
- ☆109Updated 3 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆130Updated 3 years ago
- Fully differentiable RL environments, written in Ivy.☆66Updated 2 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆208Updated last year
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 3 years ago
- Probabilistic Solution of Differential Equations☆13Updated 3 years ago