☆18Apr 17, 2026Updated last month
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 last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆60Aug 4, 2022Updated 3 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 3 years ago
- Conservative Q learning in Jax☆57Feb 7, 2023Updated 3 years ago
- ☆58Jan 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
- 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☆20Mar 10, 2021Updated 5 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆22Apr 17, 2024Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- ☆55Feb 28, 2024Updated 2 years ago
- Homework 3 for Berkeley CS 280: our version of the MIT Mini Places challenge☆12Mar 5, 2016Updated 10 years ago
- ☆18May 30, 2023Updated 3 years ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆163Dec 16, 2025Updated 5 months 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
- Author's PyTorch implementation of ICML'23 paper "Policy Regularization with Dataset Constraint for Offline Reinforcement Learning" for D…☆18Nov 8, 2024Updated last year
- ☆21Mar 19, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated last year
- Reinforcement learning library in JAX.☆102Oct 22, 2023Updated 2 years ago
- Conservative Q Learning on top of SAC☆138Oct 15, 2022Updated 3 years ago
- Gym implementation of connector to Deepmind lab☆12Mar 26, 2019Updated 7 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 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 last year
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆13Jul 11, 2022Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Building blocks for productive research☆72Mar 26, 2026Updated 2 months ago
- HPO and Architecture Benchmarking for RL: Dynamically, Reactive and Efficient☆30Mar 16, 2026Updated 2 months ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆161Nov 1, 2022Updated 3 years ago
- flexible meta-learning in jax☆16Oct 19, 2023Updated 2 years ago
- ☆23Aug 19, 2022Updated 3 years ago
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆66Dec 19, 2025Updated 5 months ago