☆18Apr 17, 2026Updated 4 months ago
Alternatives and similar repositories for transformer-exercises-jax
Users that are interested in transformer-exercises-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
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.☆22Jun 19, 2024Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆62Aug 4, 2022Updated 4 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Code for "Interactive Task Planning with Language Models"☆34Jan 12, 2026Updated 7 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 3 years ago
- Conservative Q learning in Jax☆58Feb 7, 2023Updated 3 years ago
- ☆57Jan 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 4 years ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆22Apr 17, 2024Updated 2 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- A TF2.0 implementation of RL baselines.☆10Sep 24, 2021Updated 4 years ago
- Open source code combining implementations of Upside Down Reinforcement Learning and Reward Conditioned Policies☆19Mar 10, 2021Updated 5 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 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 …☆42Mar 26, 2024Updated 2 years ago
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 3 years ago
- ☆20May 30, 2023Updated 3 years ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆165Dec 16, 2025Updated 8 months ago
- Author's PyTorch implementation of ICML'23 paper "Policy Regularization with Dataset Constraint for Offline Reinforcement Learning" for D…☆17Nov 8, 2024Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆21Mar 19, 2024Updated 2 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- Conservative Q Learning on top of SAC☆142Oct 15, 2022Updated 3 years ago
- New version: https://github.com/chungmin99/pyroki☆20Apr 13, 2025Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated 2 years ago
- Gym implementation of connector to Deepmind lab☆13Mar 26, 2019Updated 7 years ago
- Canonical Factors for Hybrid Neural Fields @ ICCV 2023☆106Jan 21, 2025Updated last year
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆13Jul 11, 2022Updated 4 years ago
- Building blocks for productive research☆72Mar 26, 2026Updated 5 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 4 years ago
- [ICML2026] Official JAX code for Emergence of Exploration in Policy Gradient Reinforcement Learning via Retrying☆16Jul 3, 2026Updated last month
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆161Nov 1, 2022Updated 3 years ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated 2 years ago
- ☆23Aug 19, 2022Updated 4 years ago
- Async isort plugin for Vim + Neovim☆23Aug 31, 2022Updated 3 years ago