louiskirsch / vsml-neurips2021

Related projects: ⓘ

• enajx / HyperNCA
  ☆34Updated 2 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated 11 months ago
• danijar / ninjax
  General Modules for JAX
  ☆57Updated last month
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆48Updated 9 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 2 years ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆52Updated last year
• ElisevanderPol / PRAE
  Plannable Approximations to MDP Homomorphisms: Equivariance under Actions
  ☆26Updated 4 years ago
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆63Updated 2 weeks ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• google-deepmind / affordances_option_models
  ☆22Updated 2 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆36Updated last year
• sebastianrisi / ga-world-models
  ☆20Updated 5 years ago
• AllanYangZhou / universal_neural_functional
  ☆46Updated 7 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆29Updated 2 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆36Updated 5 months ago
• EmptyJackson / groove
  Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…
  ☆22Updated 2 months ago
• google-deepmind / enn_acme
  ☆29Updated 2 years ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆82Updated 9 months ago
• kenjyoung / mctx_learning_demo
  ☆46Updated last year
• DramaCow / jaxued
  ☆56Updated 3 weeks ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆39Updated 2 years ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code
  ☆20Updated 2 weeks ago
• astanic / crafter-ood
  ☆20Updated last year
• lowrollr / mctx-az
  Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree
  ☆14Updated 8 months ago
• twitter-research / hyperbolic-rl
  ☆56Updated last year
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• neuroevobench / neuroevobench
  Neuroevolution Benchmark in JAX 🦕
  ☆36Updated 10 months ago
• radarFudan / mamba-minimal-jax
  ☆28Updated this week