search

jannerm / trajectory-transformer

Code for the paper "Offline Reinforcement Learning as One Big Sequence Modeling Problem"
https://trajectory-transformer.github.io
MIT License
464 stars 65 forks source link

Jax code #15

Open sglucas opened 1 year ago

sglucas commented 1 year ago

Hi,

May I ask is it possible to provide the code based on jax.

Best