☆19Apr 17, 2026Updated 6 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:
- Omnidirectional fisheye visual SLAM ROS package☆16Jun 14, 2020Updated 6 years ago
- ☆15Sep 7, 2022Updated 4 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆63Aug 4, 2022Updated 4 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆23Jun 19, 2024Updated 2 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 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.
- Offline RL experiments☆15Oct 1, 2022Updated 4 years ago
- Repo for the paper "Atari-5: Distilling the Arcade Learning Environment down to Five Games"☆12Oct 5, 2022Updated 4 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
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- A simple and easy to use implementation of the soft actor-critic algorithm.☆15Sep 2, 2022Updated 4 years ago
- Conservative Q learning in Jax☆58Feb 7, 2023Updated 3 years ago
- Gym wrapper for Vizdoom environments☆12Dec 14, 2018Updated 7 years ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆22Apr 17, 2024Updated 2 years ago
- Generic EKF, with support for non-Euclidean manifolds☆25Apr 6, 2022Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 3 years ago
- A set of environments utilizing pybullet for simulation of robotic manipulation tasks.☆28Mar 8, 2021Updated 5 years ago
- ☆58Jan 20, 2023Updated 3 years ago
- How Post-Training Shapes Biological Reasoning Models☆17Jun 16, 2026Updated 4 months ago
- Homework 3 for Berkeley CS 280: our version of the MIT Mini Places challenge☆12Mar 5, 2016Updated 10 years ago
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 3 years ago
- An Agent-Environment interface with Jax transformation compatibility☆12Nov 17, 2023Updated 2 years ago
- ☆162Oct 12, 2021Updated 5 years ago
- ☆20May 30, 2023Updated 3 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- NeuralDMD for Imaging from Sparse Observations☆12Jul 9, 2025Updated last year
- A relay between Gymnasium and any software☆11Aug 11, 2025Updated last year
- ☆21Mar 19, 2024Updated 2 years ago
- RL and MARL with Flax NNX☆16Sep 7, 2026Updated last month
- Author's PyTorch implementation of ICML'23 paper "Policy Regularization with Dataset Constraint for Offline Reinforcement Learning" for D…☆18Nov 8, 2024Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated 2 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- A prettier rocm-smi output with color-coded GPU stats☆11Feb 19, 2026Updated 8 months ago
- Gym implementation of connector to Deepmind lab☆13Mar 26, 2019Updated 7 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- [ICML2026] Official JAX code for Emergence of Exploration in Policy Gradient Reinforcement Learning via Retrying☆17Oct 17, 2026Updated last week
- New version: https://github.com/chungmin99/pyroki☆20Apr 13, 2025Updated last year
- Benchmarked implementations of Offline RL Algorithms.☆77Mar 4, 2025Updated last year
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆162Nov 1, 2022Updated 3 years ago