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
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• jamesvuc / jax-bayes
  Bayesian inference with Python and Jax.
  ☆32Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆109Updated 2 years ago
• google-deepmind / jaxline
  ☆156Updated last year
• google-deepmind / tf2jax
  ☆114Updated last week
• 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
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆92Updated last year
• chandar-lab / RLHive
  ☆101Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 9 months ago
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆101Updated last year
• google-deepmind / enn
  ☆311Updated 2 months ago
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆83Updated last year
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• 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
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 4 months ago
• aleatory-science / deep-probprog-course
  Deep Probabilistic Programming Course @ DIKU
  ☆57Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• google-deepmind / enn_acme
  ☆31Updated 2 years ago
• google-research / hyperbo
  Pre-trained Gaussian processes for Bayesian optimization
  ☆90Updated 3 weeks ago
• MichaelArbel / GeneralizedEBM
  ☆53Updated 9 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆87Updated last year
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• microsoft / bayesianize
  Bayesianize: A Bayesian neural network wrapper in pytorch
  ☆88Updated 11 months ago