Really cool stuff, but really surprised no RL being used here, https://github.com/rail-berkeley/serl seems to be the way forward imho