Alternatives and similar repositories for minGPT-flax

Users that are interested in minGPT-flax are comparing it to the libraries listed below

• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 3 years ago
• danijar / elements
  Building blocks for productive research
  ☆60Updated last month
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 10 months ago
• danijar / ninjax
  General Modules for JAX
  ☆67Updated this week
• orybkin / lexa-benchmark
  ☆42Updated 3 years ago
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 4 months ago
• facebookresearch / entity-factored-rl
  Source code for the paper "Policy Architectures for Compositional Generalization in Control"
  ☆30Updated 3 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 years ago
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆20Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 2 years ago
• microsoft / segar
  Sandbox environment for generalizable agent research
  ☆26Updated 3 years ago
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 3 years ago
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆58Updated 3 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 9 months ago
• albertwilcox / mcac
  Author implementation of Monte Carlo Augmented Actor Critic in PyTorch
  ☆17Updated 2 years ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆58Updated last year
• google-deepmind / affordances_option_models
  ☆23Updated 3 years ago
• google-deepmind / csuite
  ☆44Updated 11 months ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆56Updated 2 years ago
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆68Updated last year
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 3 years ago
• jannerm / gamma-models
  Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"
  ☆45Updated last year
• rll-research / cic
  CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery
  ☆81Updated 3 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆14Updated last year
• denisyarats / proto
  Proto-RL: Reinforcement Learning with Prototypical Representations
  ☆83Updated 3 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆35Updated last year
• zchuning / latco
  Model-Based Reinforcement Learning via Latent-Space Collocation.
  ☆33Updated 2 years ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆31Updated 2 years ago
• ElisevanderPol / PRAE
  Plannable Approximations to MDP Homomorphisms: Equivariance under Actions
  ☆30Updated 5 years ago