Alternatives and similar repositories for neural_testbed

Users that are interested in neural_testbed are comparing it to the libraries listed below

• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 3 years ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• google-deepmind / tf2jax
  ☆114Updated 3 weeks ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆83Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆59Updated 3 years ago
• ivy-llc / gym
  Fully differentiable RL environments, written in Ivy.
  ☆65Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆87Updated 2 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago
• chandar-lab / RLHive
  ☆101Updated last year
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆56Updated 2 years ago
• PrincetonLIPS / RandomizedAutomaticDifferentiation
  Experiment code for "Randomized Automatic Differentiation"
  ☆67Updated 4 years ago
• CampusAI / Hamiltonian-Generative-Networks
  Re-implementation of Hamiltonian Generative Networks paper
  ☆33Updated 2 years ago
• gehring / fax
  ☆80Updated 3 years ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆37Updated last month
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 2 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• desi-ivanova / idad
  Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods
  ☆19Updated 3 years ago
• AllanYangZhou / metalearning-symmetries
  Experiments for Meta-Learning Symmetries by Reparameterization
  ☆56Updated 4 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆107Updated 11 months ago