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 / ninjax
  General Modules for JAX
  ☆69Updated last month
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆43Updated 11 months ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 10 months ago
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆20Updated 3 years ago
• danijar / elements
  Building blocks for productive research
  ☆62Updated 2 months ago
• google-deepmind / csuite
  ☆46Updated last year
• 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
• microsoft / segar
  Sandbox environment for generalizable agent research
  ☆25Updated 3 years ago
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 5 months ago
• facebookresearch / entity-factored-rl
  Source code for the paper "Policy Architectures for Compositional Generalization in Control"
  ☆30Updated 3 years ago
• orybkin / lexa-benchmark
  ☆42Updated 3 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆38Updated last year
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆31Updated 2 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆29Updated 3 years ago
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆68Updated last year
• qgallouedec / lge
  ☆31Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 3 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 years ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆57Updated 2 years ago
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 4 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆15Updated 2 years ago
• denisyarats / proto
  Proto-RL: Reinforcement Learning with Prototypical Representations
  ☆85Updated 3 years ago
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 3 years ago
• rll-research / cic
  CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery
  ☆81Updated 3 years ago
• scottemmons / rvs
  Reinforcement Learning via Supervised Learning
  ☆71Updated 3 years ago
• jannerm / gamma-models
  Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"
  ☆45Updated 2 years ago
• albertwilcox / mcac
  Author implementation of Monte Carlo Augmented Actor Critic in PyTorch
  ☆17Updated 3 years ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆82Updated 3 years ago