GitHub repository: too-z/jax_dt
Description: Minimal Decision Transformer Implementation written in Jax (Flax).