Minimal Decision Transformer Implementation written in Jax (Flax).
Primary LanguagePythonMIT LicenseMIT