☆18Mar 10, 2026Updated last week
Alternatives and similar repositories for transformer-exercises-jax
Users that are interested in transformer-exercises-jax are comparing it to the libraries listed below
Sorting:
- ☆15Sep 7, 2022Updated 3 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆59Aug 4, 2022Updated 3 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 2 years ago
- Code for "Interactive Task Planning with Language Models"☆33Jan 12, 2026Updated 2 months ago
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 2 years ago
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 2 years ago
- Conservative Q learning in Jax☆57Feb 7, 2023Updated 3 years ago
- ☆53Jan 20, 2023Updated 3 years ago
- A simple and easy to use implementation of the soft actor-critic algorithm.☆15Sep 2, 2022Updated 3 years ago
- Generic EKF, with support for non-Euclidean manifolds☆25Apr 6, 2022Updated 3 years ago
- A TF2.0 implementation of RL baselines.☆10Sep 24, 2021Updated 4 years ago
- Like `typing._eval_type`, but lets older Python versions use newer typing features.☆16Dec 2, 2025Updated 3 months ago
- Open source code combining implementations of Upside Down Reinforcement Learning and Reward Conditioned Policies☆19Mar 10, 2021Updated 5 years ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆23Apr 17, 2024Updated last year
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- Homework 3 for Berkeley CS 280: our version of the MIT Mini Places challenge☆12Mar 5, 2016Updated 10 years ago
- ☆55Feb 28, 2024Updated 2 years ago
- Offline RLHF codebase implementation for "Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human …☆41Mar 26, 2024Updated last year
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆163Dec 16, 2025Updated 3 months 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
- ☆20Mar 19, 2024Updated 2 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- Gym implementation of connector to Deepmind lab☆12Mar 26, 2019Updated 6 years ago
- Conservative Q Learning on top of SAC☆137Oct 15, 2022Updated 3 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Building blocks for productive research☆69Jan 12, 2026Updated 2 months ago
- New version: https://github.com/chungmin99/pyroki☆19Apr 13, 2025Updated 11 months ago
- Canonical Factors for Hybrid Neural Fields @ ICCV 2023☆106Jan 21, 2025Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated last year
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆13Jul 11, 2022Updated 3 years ago
- HPO and Architecture Benchmarking for RL: Dynamically, Reactive and Efficient☆28Updated this week
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 8 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆160Nov 1, 2022Updated 3 years ago
- flexible meta-learning in jax☆16Oct 19, 2023Updated 2 years ago