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
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆20Updated 3 years ago
• danijar / ninjax
  General Modules for JAX
  ☆72Updated 4 months ago
• microsoft / segar
  Sandbox environment for generalizable agent research
  ☆26Updated 3 years ago
• facebookresearch / entity-factored-rl
  Source code for the paper "Policy Architectures for Compositional Generalization in Control"
  ☆30Updated 3 years ago
• danijar / elements
  Building blocks for productive research
  ☆67Updated 2 weeks ago
• orybkin / lexa-benchmark
  ☆42Updated 3 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated last year
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 8 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• 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
• 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
• google-deepmind / csuite
  ☆46Updated last year
• rll-research / cic
  CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery
  ☆83Updated 3 years ago
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 3 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆46Updated 3 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆30Updated 4 years ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆57Updated 3 years ago
• 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
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years ago
• denisyarats / proto
  Proto-RL: Reinforcement Learning with Prototypical Representations
  ☆86Updated 3 years ago
• scottemmons / rvs
  Reinforcement Learning via Supervised Learning
  ☆72Updated 3 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆16Updated 2 years ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆122Updated last year
• jannerm / gamma-models
  Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"
  ☆46Updated 2 years ago
• yobibyte / amorpheus
  My Body Is A Cage
  ☆41Updated 4 years ago
• ademiadeniji / irm
  Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)
  ☆42Updated 2 years ago