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
  ☆44Updated last year
• danijar / elements
  Building blocks for productive research
  ☆65Updated 4 months ago
• danijar / ninjax
  General Modules for JAX
  ☆71Updated 3 months 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
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆58Updated 3 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated last year
• facebookresearch / entity-factored-rl
  Source code for the paper "Policy Architectures for Compositional Generalization in Control"
  ☆30Updated 3 years ago
• microsoft / segar
  Sandbox environment for generalizable agent research
  ☆25Updated 3 years ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆32Updated 2 years ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 7 months ago
• rll-research / cic
  CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery
  ☆83Updated 3 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆45Updated 3 years ago
• google-deepmind / zipfian_environments
  ☆28Updated 3 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆30Updated 4 years ago
• orybkin / lexa-benchmark
  ☆42Updated 3 years ago
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆20Updated 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
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 years ago
• lili-chen / SEER
  Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.
  ☆21Updated 4 years ago
• google-deepmind / csuite
  ☆46Updated last year
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated last year
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆82Updated 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
• qgallouedec / lge
  ☆32Updated last year
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• facebookresearch / denoised_mdp
  Open source code for paper "Denoised MDPs: Learning World Models Better Than the World Itself"
  ☆136Updated 2 years ago
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 3 years ago