GPT implementation in Flax
☆18Jan 8, 2022Updated 4 years ago
Alternatives and similar repositories for toy-transformers-jax
Users that are interested in toy-transformers-jax are comparing it to the libraries listed below
Sorting:
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- ☆15Sep 7, 2022Updated 3 years ago
- Open source code combining implementations of Upside Down Reinforcement Learning and Reward Conditioned Policies☆19Mar 10, 2021Updated 4 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆59Aug 4, 2022Updated 3 years ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆23Apr 17, 2024Updated last year
- A TF2.0 implementation of RL baselines.☆10Sep 24, 2021Updated 4 years ago
- Gym implementation of connector to Deepmind lab☆12Mar 26, 2019Updated 6 years ago
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- A simple and easy to use implementation of the soft actor-critic algorithm.☆15Sep 2, 2022Updated 3 years ago
- ☆52Jan 20, 2023Updated 3 years ago
- Maximum Entropy-Regularized Multi-Goal Reinforcement Learning (ICML 2019)☆24May 30, 2019Updated 6 years ago
- ☆55Feb 28, 2024Updated 2 years ago
- Conservative Q learning in Jax☆57Feb 7, 2023Updated 3 years ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆13Jul 11, 2022Updated 3 years ago
- 3D learning environment with rigid body simulation for Linux/MacOSX☆14Dec 24, 2021Updated 4 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- Corax: Core RL in JAX☆40Feb 22, 2024Updated 2 years ago
- MImE - Manipulation Imitation Environments☆14Feb 1, 2022Updated 4 years ago
- Evaluating different engineering tricks that make RL work☆15Jun 3, 2021Updated 4 years ago
- PALMER: Perception-Action Loop with Memory for Long-Horizon Planning, NeurIPS 2022☆15Dec 12, 2022Updated 3 years ago
- Implementation for "ROLL: Visual Self-Supervised Reinforcement Learning with Object Reasoning", CoRL 2020☆16Jun 22, 2022Updated 3 years ago
- Code Released for NeurIPS 2018 paper: Synthesized Policies for Transfer and Adaptation across Tasks and Environments☆16Apr 17, 2019Updated 6 years ago
- Reinforcement Learning via Supervised Learning☆72May 16, 2022Updated 3 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- ☆18May 30, 2023Updated 2 years ago
- ☆20Mar 19, 2024Updated last year
- Motion imitation with deep reinforcement learning.☆13Jul 24, 2019Updated 6 years ago
- The MAGICAL benchmark suite for robust imitation learning (NeurIPS 2020)☆78Dec 5, 2023Updated 2 years ago
- Episodic Control☆22Sep 20, 2022Updated 3 years ago
- ☆19Jul 18, 2021Updated 4 years ago
- Author's PyTorch implementation of ICML'23 paper "Policy Regularization with Dataset Constraint for Offline Reinforcement Learning" for D…☆18Nov 8, 2024Updated last year
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Jul 6, 2023Updated 2 years ago
- RAD: Reinforcement Learning with Augmented Data (code for procgen experiments)☆18Mar 29, 2021Updated 4 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- In Progress : State of the art Distributed Distributional Deep Deterministic Policy Gradient algorithm implementation in pytorch.☆19Jun 15, 2018Updated 7 years ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Oct 12, 2023Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago