DecisionMamba

Decision Mamba is an implementation of decision transformers utilizing the Mamba architecture, optimized for both offline and online reinforcement learning (RL). Its key features include a convolutional mode for efficient batch training and a recurrent mode designed for rapid inference, making it suitable for fast online training scenarios. The tool aims to facilitate model training using the Mujoco environment while offering straightforward installation guidelines.