On GitHub
On COLAB
This tutorial goes over Task objects, how they work and how to build a custom subclass to implement your own task design.
1. Initialization of the Task subclass
2. Generating inputs
a. Input dictionary
b. Targets
c. Initial states