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
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆45Updated last year
• orybkin / lexa-benchmark
  ☆42Updated 3 years ago
• danijar / elements
  Building blocks for productive research
  ☆67Updated this week
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 8 months ago
• danijar / ninjax
  General Modules for JAX
  ☆72Updated 3 months ago
• microsoft / segar
  Sandbox environment for generalizable agent research
  ☆25Updated 3 years ago
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆59Updated 3 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated last year
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆57Updated 3 years ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆32Updated 2 years ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆58Updated 3 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆46Updated 3 years ago
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆20Updated 3 years ago
• google-deepmind / csuite
  ☆46Updated last year
• facebookresearch / entity-factored-rl
  Source code for the paper "Policy Architectures for Compositional Generalization in Control"
  ☆30Updated 3 years ago
• scottemmons / rvs
  Reinforcement Learning via Supervised Learning
  ☆72Updated 3 years ago
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 4 years ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 4 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆30Updated 4 years ago
• rll-research / cic
  CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery
  ☆83Updated 3 years ago
• google-deepmind / nao_top10
  ☆19Updated 2 years ago
• google-deepmind / affordances_option_models
  ☆23Updated 4 years ago
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 3 years ago
• denisyarats / proto
  Proto-RL: Reinforcement Learning with Prototypical Representations
  ☆85Updated 3 years ago
• jannerm / gamma-models
  Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"
  ☆45Updated 2 years ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆17Updated last year
• evgenii-nikishin / omd
  JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
  ☆44Updated 4 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year