brentyi / minGPT-flaxLinks
GPT implementation in Flax
☆18Updated 3 years ago
Alternatives and similar repositories for minGPT-flax
Users that are interested in minGPT-flax are comparing it to the libraries listed below
Sorting:
- Generalised UDRL☆37Updated 3 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 months ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 8 months ago
- Sandbox environment for generalizable agent research☆25Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆28Updated 2 years ago
- Author implementation of Monte Carlo Augmented Actor Critic in PyTorch☆17Updated 2 years ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆57Updated last year
- ☆44Updated 9 months ago
- General Modules for JAX☆66Updated 3 months ago
- flexible meta-learning in jax☆14Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆57Updated 2 years ago
- Latent World Models For Intrinsically Motivated Exploration | Official repository☆22Updated 4 years ago
- Fast reinforcement learning research☆61Updated 7 months ago
- Building blocks for productive research☆59Updated 5 months ago
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆16Updated last year
- Vectorization techniques for fast population-based training.☆56Updated 2 years ago
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 4 years ago
- ☆15Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 11 months ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- Official Implementation of SFM and the baselines in Jax.☆19Updated last month
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Updated 2 years ago
- ☆42Updated 3 years ago
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆44Updated last year
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆17Updated 2 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆33Updated 2 years ago