Imported from https://github.com/for-ai/TD